From fc7a5a27cd0037907e52af412252f7203f071e39 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Mon, 9 Jan 2023 16:09:33 +0900 Subject: [PATCH] from local --- .../.train_adapter.py.swp | Bin 81920 -> 86016 bytes .../train_adapter.py | 6 +++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp index 785b91fa2f0b9c78eea583782e07fe2eca8bd036..3d850fbf8b5f5214834c44fe62ea2b604a3d1661 100644 GIT binary patch delta 590 zcmY+=Pe_w-9LMqRZ$3|B)A{Vdd{%mJD8+0s<|PxG9@B$elnzFSZ5G1gES-W@YzE!a z=P+;niO7Q&H)&dQ2!fzf2uV;E1uIHy7k4qJ_s|Y~;l=OA_w)V1k8dWUXBKtqR`hyz zPp~Z#70NG?e*G$Lzgx5R4ONq`@Q)}JbMBb|^K?06>RZ}yi+rH<1idK7mvF@BCp^O#I?;g`YETJe68VYuSW$&lQ$5z4ok+;l{4o3Gin5$y#0-Qd2kqg+ zz;N>R=#9pBDn4M3j*mCqbatNBmh-y*^1EXb$)Uu_NOs0jXPm{7+nt;k8L}^gvImwr z>y_Y1H94~-s}KpOtx$R}A+qZcS@s^U@85Pe4fcZ;L~&RxvV{fMXv1Z+!Ur8kK9OI@ zA%zw+pj0LD1M67D98&12l6JYu I!c_hGUrJ|mPXGV_ delta 275 zcmXBOp-TgC7{>AE-S*ppxZTOzVGu70!dwz;F3kyI(_uWV2D{)PYg%OC>M)2+?iR+- zG%QBNKR^Vv7%VPW^eaB_fhRoSiTf^&T`g*Cw>QJpdPBsMs@cunc$C&YDmF97dk1Dw z@6ZnhSAM5_{v2ju;d8poG_(JGl=z;#AA2R~z9^lbi8VO9IO!TkXk#5M%wrZ)_$f$Z zbkW2ziule;ceup`y69jBD_Fo=PI}F0Nh#$Fr`X3FDwy!|3s3m_oLcf=Rvlh$*se=P PTNd>ClhUvr*yG|ax92m^ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py index 1d5e0f341..5f0f9db66 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py @@ -1202,7 +1202,11 @@ def train_one_epoch( if wb is not None and rank == 0: numel = 1 / (params.accum_grads * valid_info["utterances"]) - wb.log({"valid/loss": valid_info["loss"]*numel}) + #wb.log({"valid/loss": valid_info["loss"]*numel}) + wb.log({"valid/loss": numel*(valid_info["simple_loss"] + +valid_info["pruned_loss"] + +valid_info["ctc_loss"] + )}) wb.log({"valid/simple_loss": valid_info["simple_loss"]*numel}) wb.log({"valid/pruned_loss": valid_info["pruned_loss"]*numel}) wb.log({"valid/ctc_loss": valid_info["ctc_loss"]*numel})