From 62560a861b952ffb6b6bedda21a8d16e6feda05b Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 15:25:58 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 110592 -> 110592 bytes
 .../train.py                                  |   5 +++--
 2 files changed, 3 insertions(+), 2 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 6d45790d97a470e80c98bc77fadbb4c8286e034c..fc947f5e6ad0e325a9699fba499c245f625f01c8 100644
GIT binary patch
delta 319
zcmXZXu}i~X5C!nNez8s3N;OEUPO+ufNlG^ZqDu$Su2L#eFrbKHQooE91Et`QEkb4y
z>f$09G6>?NgU*hUKS8MC>MMHi<KUL}&P&o>k}mGocQ#wb>Pnq3GeqO+l~pO+7dtwe
zQBM!{roNb{7HLk2^}Is#D2N=W!Lo=Ny>AX|X)Ic<)jhGLEStt;b&pOD?A`L@8eGdb
z=*x{k+l|{U-{q#4GDKDNFVGBUJwCoj<%t-cb3`}bz#3#h0~J29MDOqhPcVQcq~KE{
zdVvQR!Z}#bgaw#~A4Dc_3nMs#7Hq2|&?jQW$b{=#tj33zJQf1w!^Gz~<^PLP`08^Z
J?)W^a`~u^`M*{!=

delta 293
zcmXBOu}cDR90l<2SIyJR&>M7c@E6e(i8Xb&)sjl9Tn7zjb`gO;y}MEBK#+oJxxl5d
zni`bfa9dOKcQ^z!wzO144|?!f-+RF<8O)M}cC}UC6-K2>NRH@Z<>ugfaayo+HdJ2^
z_vdOcBB|27BJ00#q6zdsf|*bo{ZAg>U+_*SJrS<@YvI7V9h!b*`uv#5IvSCa`mZ*|
z`G8N(wibvqc+C(^;S`$CfI5_+2w7lwo+lc?HmpMme$qrQ@C0`-0t@zF6PDl|vu~i_
r8m<6WR4mcZN&-1{gzX-A75}j7Nw+O+w<GLsCn}%wS3I&V`C<7FhnzoT

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 5b3579f79..293589a02 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -1105,10 +1105,10 @@ def train_one_epoch(
                     )
             
             if wb is not None and rank == 0:
+                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/loss": loss_info["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")
@@ -1134,6 +1134,7 @@ def train_one_epoch(
         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})
+        wb.log({"valid/ctc_loss": valid_info["ctc_loss"]*numel})
 
     loss_value = tot_loss["loss"] / tot_loss["utterances"]
     params.train_loss = loss_value