From cb0aa1247d382264d28e8f04159e740a086899df Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 10 Dec 2022 14:04:01 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 102400 -> 102400 bytes .../train.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) 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 70ce09f170f9896a2bdee42c6c64c59ea3a2e105..f070be63a4879017209b03a4c12a17c37b156818 100644 GIT binary patch delta 211 zcmWN=KMMh290%~<)1CX{PbqZ_idUe!EDR>4j%>&NxNennvbnQdk20!TZ!jxKw^_(5 z@Cq>7YzE&?eZoWu6UCUAV|!R{H%x*eQLD6bt?$}68|)KgxJtxhJ!H#3%@e(IL`PiV z0Q+cT4a-F+k#b6 UPq2riEqvdb^QoJ5w(KhX0m8Q@mjD0& delta 211 zcmWN=y$bIq^ch!tz(^jof7D?6YJpyvci80; z7dgu*zUjc0n_OX&!FLU~@{zYZ%H8i9g0Ny0|4eHFaQ7m 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 f47f84049..82ffc8b50 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1089,7 +1089,7 @@ def train_one_epoch( 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"] + loss_value = tot_loss["loss"] / tot_loss["utterances"] params.train_loss = loss_value if params.train_loss < params.best_train_loss: params.best_train_epoch = params.cur_epoch