From 53ce89117ac4e12527dbd8a5904ccbbfa51ccea4 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Mon, 12 Dec 2022 16:21:04 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 69632 -> 69632 bytes
 .../train.py                                  |   8 ++++----
 2 files changed, 4 insertions(+), 4 deletions(-)

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 cbb0b3b8a9cca669be06209ebbfb8b97347c1e72..85453e084d8c2d2952e66377ee22667eebc94424 100644
GIT binary patch
delta 563
zcma*h&nrYx6u|K_JP&V%8KqY&j9*!5B+7;|7A$N=ky67bWBizh!h}4fS($};H8yC-
z%*IMXNi3!;>{uwtLQ;N}AO8VgKmGuxKKGn^?!BkGlu(uuiXQNEcXXCF)p~?*i7a2n
z2d_@|^f$j6q2tN2wEj|a!YNXT%R-SDR?rL=9tuRxafDs0ViKe1MG-z7BHLKUBKlB<
zVmR?)7fE3pHMrG8HnD+#CZ@DdIB=cMMI^C<IZR;~eQ@J0kH?5%5Iv|s2|jZ<#{u?X
zVF6yWq7cuj$P+S{!I)}_bW!+F2Pf`xL{4ywI3k!u8+5!V48jG@k;DS#(F_{|XW4uf
zG5Ar9D!gVfA5oaaEQ<)i2M->ttxWyeyUmtoeGF)g<$p0|1cRY)ur?I_Psegk>z3bC
e(@G%fux>`|Rx0fM$*S?2hm3~refrR7PkaF`;A<ZM

delta 576
zcmZ|JJt)Kh7{~G7aSng)x+qFs2hzC^84Pj*d0B9SxN}|xb({`~<!%w1^e>XqoiZ2<
zGGJ6*OUgnbD;7)?lCPboK2J}--|~bb)^Nm{)$epSx8>J4oMO@?`_a-s?dEpYv)A?;
zJz7koFDK+llfF}>HMr1-w-o6VYnVe21871iijaY`WN8=O=tK?dxJr`Fae{StP=b3y
zT80~4$T1Ysd!lrRMR?GQ9#o<X`LN?UL0W+uF61H&H}TRXcCmsH_|bxDeA=WB93zSd
zrr|)bEhHsz7<h`~W2|Bl6KIDM75K7B7g$FW^O!>~s!@fvSZN=72*V8*vT+w9-C_fi
zs7I}xRf7T;c(h0lX7r$H`Nd*MFkgL!nLcGTy&>C9YyWI=X2qB<=o=Y#Oa+4eLGSqR
c%#e9G+WvQ3Xf_1O|Em@Pd4E(pTY-+q54P)bQUCw|

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 a3109aa97..aa73a31a9 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1135,10 +1135,10 @@ def train_one_epoch(
                     )
             
             if wb is not None and rank == 0:
-                wb.log({"train/loss": loss_info["loss"]*numel/params.world_size})
-                wb.log({"train/simple_loss": loss_info["simple_loss"]*numel/params.world_size})
-                wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel/params.world_size})
-                wb.log({"train/ctc_loss": loss_info["ctc_loss"]*numel/params.world_size})
+                wb.log({"train/loss": loss_info["loss"]*numel})
+                wb.log({"train/simple_loss": loss_info["simple_loss"]*numel})
+                wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel})
+                wb.log({"train/ctc_loss": loss_info["ctc_loss"]*numel})
 
 #if batch_idx % params.valid_interval == 0 and not params.print_diagnostics:
     logging.info("Computing validation loss")