From d0ddca6653917268181d15a5559f494d8f0cccc8 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Fri, 13 Jan 2023 17:13:22 +0900
Subject: [PATCH] from local

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

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.train.py.swp
index 271c4c4ad051c132d9f6f95de328e9ff058ef2b9..70c5e80f9da74417ebf8b474c8c90420cd4e80c3 100644
GIT binary patch
delta 501
zcmZo@U};#uvVlo}k$*F@052nB;$%TVaX|qF2;hOzU7IBZkMnI(U}@mJ3KS}30un$q
ziJJu%lrtJNGBYr^1Mz>L@OB^uNpAz<nLykG#PL9E0mNEBEC9p@7$@IYIC*lSzInYu
zd6HgEe!51rQb|!_W}bd>NpgHnesQsql>&&0&&*59k5+<<#@1>wD4+o}6AFq-^HNi=
z7=kPTGp8QeEHs0PGjj`aQn46>ECDkJk2#p$0gHfj${^{>OalgnLT0f-UVe#!Uw&Td
z=6Q>gVq6nb7#KeCGBDWkLBbpsCI9&u7<K`13lIkZu`v*f0r7vJx;sF;6^NGuaW4?t
UZccpggLiU*6UXLf3wJXD0NH(mTmS$7

delta 148
zcmZozz|zpbvVlo}k#94z052nB`eZ>t@y#BBJNPC(P}rov(!i_6z`$^i5l9FyFr;r5
zTu{!a8_mqXpbEsFnHU(Z0P%JpUI4_kKpY0d9zd)I#2{^VCogQ2p7?-oa)J}b<P!^9
pC)Xa9n>=AH%j5(5*(TpTEIHYKIqP%*e#Q@z8<z2GUb{Fr1^|e8D(C<J

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
index cd2ec971e..c143d98a4 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/train.py
@@ -901,6 +901,12 @@ def train_one_epoch(
                 )
                 tot_loss.write_summary(tb_writer, "train/tot_", params.batch_idx_train)
 
+            if wb is not None:
+                wb.log({"train/loss": loss_info["loss"]})
+                wb.log({"train/simple_loss": loss_info["simple_loss"]})
+                wb.log({"train/pruned_loss": loss_info["pruned_loss"]})
+                wb.log({"train/ctc_loss": loss_info["ctc_loss"]})
+
         if batch_idx > 0 and batch_idx % params.valid_interval == 0:
             logging.info("Computing validation loss")
             valid_info = compute_validation_loss(