From 1754bf0cd79cb82e4f4d06af4e8dcc97c9b2ee1b Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 16:12:25 +0900 Subject: [PATCH] from local --- .../.prompt_tuning.py.swp | Bin 90112 -> 90112 bytes .../prompt_tuning.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp index c45141abe409f4f5b92730f2cb597c35dcbd55b0..f6a3372160329d1156d05da83e912c34bc12de03 100644 GIT binary patch delta 184 zcmZoTz}j$tRV>LM%+puFQqO<^2m}}yzO$RABrMq|CMC$2G1*X1o{@8Npy1|nyk(3G z3`WclH5r=)AAHU?^WkG)=mBDoHcpUsAb8Br!0-r&yMfpfh&_N<2Z-+g)!hcG-|iT| z=*BmB-$I`0b)1YGlRxOOY<}@jm2tW?3nS0;XYq_uvILM%+puFQqO<^2m}}yl-NyEk{53jlM-aioNOp4&&aVkP;m1(USCEA z23}@}n#|3D4?gD`Ao$PEz_0>{(||Y?h&_S$Cs54~u-fg80gU#1 T6CapNZ-`^$**+tV(VGzf$Y>pe diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py index 2aabd8765..a0906b5b1 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py @@ -1063,7 +1063,7 @@ def train_one_epoch( try: loss_info.reduce(loss.device) except: pass - logging.info(loss_info) + print(loss_info) numel = params.world_size / (params.accum_grads * loss_info["utterances"]) loss *= numel ## normalize loss over utts(batch size)