From 8deeece377704d269db834795a3bf43acd4476c4 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 14:03:52 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 102400 -> 102400 bytes
 .../train.py                                  |   6 ++++++
 2 files changed, 6 insertions(+)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
index a7e3822a39c1ddd820e8c900f9ec41916afd3653..70ce09f170f9896a2bdee42c6c64c59ea3a2e105 100644
GIT binary patch
delta 375
zcmZozz}B#UO*F|M%+puFQqO<^2m}}yxP+%9|J^7mEyQTqY$&wdP>AtzE^jJ314AkY
zM1kdW!3M_Q`et4RhBzSR0^-9w3=B0u><q+CK+FxqFSr>PwgPbz5c>nM9}xQju`&=#
z1F;|wvjXu6E(V4!Aa)00Qy}I6Vg?{S#mT^M2#8kzaTgFr0&xTon**@|CxbY{bD(>k
z0r4gv?gio=AWr4j%*f)+KD{7>k!AY2PDa`7t2-H=GE0{y>E-07Yg8+hCFW$N=;!1Y
z7b{sQfH{+ImCGri$rlur=B1{@qbO6XP(@c(oS9pYlZvV=ze07oWi6u-E7<Vq`L&Fq
Tlc!W@PIioz*gn6PF_8%X-KSXt

delta 357
zcmZozz}B#UO*F|M%+puFQqO<^2m}}yJ_=1qzOYeLT8PoO*-&V^p%CNcT;4+*3=FF{
zAqtGA3pOwY*SGUBFa!XxE)f6XVPH58#8ZHHG7wJ!;tU{80OBwp_5xyFAU@B{z|aH4
zRX`jF#I8Wh55(+1{FVzS55)6;crFmv0C6f1>j1Gf5Pt_+e+h^$0x{4jo0(WFIHnhb
kFtXHVrYR&U0Kov!JjfPrPwilQ#yr_8nt%KJUdBWw01@U%X8-^I

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
index a7bea4a07..f47f84049 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1082,6 +1082,12 @@ def train_one_epoch(
         valid_info.write_summary(
             tb_writer, "train/valid_", params.batch_idx_train
         )
+    
+    if wb is not None and rank == 0:
+        numel = params.world_size / (params.accum_grads * valid_info["utterances"])
+        wb.log({"valid/simple_loss": valid_info["simple_loss"]*numel})
+        wb.log({"valid/pruned_loss": valid_info["pruned_loss"]*numel})
+        wb.log({"valid/loss": valid_info["loss"]*numel})
 
     loss_value = tot_loss["loss"] / tot_loss["frames"]
     params.train_loss = loss_value