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

---
 .../.train.py.swp                             | Bin 98304 -> 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 dc73f9efb89b8c8a326b446ba009e998cd49596b..f60604e5296e037e30c3c8b539556f079e16ee74 100644
GIT binary patch
delta 905
zcma*lJ!n%=6u|NGl9!~lO%sE$Vh10wViimRg0zK5Y!^FN6p9E^Ytp3kk@O`^QW7XO
z0iCosG+G5ePzN0xss=0KAUYTn99%?k>d;Ao!AS?H|JM%)DAWVLcQ5zcd${j%N+o;g
zuH73Ao*X!)_4tEAxJ4E{S4Zcbcxx`Dy<&AcEsDRq)zj|TVWx?LYlG}nD?2@JoliCO
zxbv%BWWhm#<Lpj9EWT(GnZ;q$$vlR+;}NbPk0BhuF4*vlDmQTmU9jM@O=J!w6mb-L
z@Iw`Og;`w2C7i`6462flJ|;F)ts)nZ!2m*VV8tS`j4I}F4LJ;>6T9(U5qXbyxQ}_5
zp{<5KrbRO2S$}0|vSur;FRrD`Z$9czrpCLbeYs3THv*gGlc{Xh*Q=Ssh;EFfPWx6@
zhW8s&6Y=EveVd#w2GW@+BOV+1yG~=`f6E23`b0Vz|3|gPL_@W1^Ox6yhsj4ZJ*ye1
zoHm#;;#$OrX_<(T&_W?C*ee?w&9Qt$_m;!@_6zmG^LjyXJBpDQO-9>vH1pIn)JDUt
zxM&|tp&y-apx!3(0S|BkGw@>@7MvpYFpU#v!LL@4pLmTsNT3@wsI4+8RwiFsL_VU5
zmzY5qzqg2d!y7E3gaXbW3>Vr^<MN+y8@F%{{pdqSb45{mny#mZ6t%KExn%zX0TQ^)

delta 340
zcmXBPF(`y_7{~GF|K96=@0B;Nx9;A-;6Gf(>x?K-ml%{up{P40WnfS@D6b3{40`2V
z6r1_H<)kpkfDDSobOwu3H<;!7=Bdx~JoWsRCp^_*sPonG^u(kYD3yisL<0Y?W;OEP
zU8;~|Jl9flQMVr?<Nl?4&@$)7o^oeMeyIOA*VVJldq-pqL&)JPEAosBL|DN*D)`YN
z?^wqYMiJnpL*y1W*uf(D@oS4b;SmowL5zKDp<>H|XqLN-NCUgrz!-*6gvOgC@`?n<
zsG*7ya&VBo)xs5y(ibUX;+{QKOk1BaO~$8=uV+<f+&ie8Xk@vt-L5y=b>(IM0J!cv
A;Q#;t

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 51a67e637..ee57ec55a 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1057,6 +1057,12 @@ def train_one_epoch(
                         cur_grad_scale,
                         params.batch_idx_train,
                     )
+            
+            if wb is not None and rank == 0:
+                wb.log({"train/simple_loss": loss_info["simple_loss"]*numel})
+                wb.log({"train/pruned_loss": loss_info["pruned_loss"]*numel})
+                wb.log({"train/loss": loss_info["loss"]*numel})
+
 
         if batch_idx % params.valid_interval == 0 and not params.print_diagnostics:
             logging.info("Computing validation loss")