From fbb80af5ec8b584cfcfe477c5bf57ca4d4bed9d0 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 25 Jan 2023 17:22:09 +0900 Subject: [PATCH] from local --- .../.train_adapter.py.swp | Bin 86016 -> 86016 bytes .../train_adapter.py | 1 + 2 files changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp index 9ebe588f2ffbf7931c4343c3562de855e9a672c3..698100a2c807b549042d933e0283d58c289f57a1 100644 GIT binary patch delta 223 zcmZozz}m2YRW!*U%+puFQqO<^2m}}yDxO_P-nvm#T97egv!USSbG*zz@%@Yt$&Af{ z5A+%9SFtfLtOVjYKpY9gnn273#7|in7)}B4N+50qVl5z+0OEHn3=D69cp4Cg0C6x7 zn*;H4pxMuWcsdXV0I?bns{-*gCZHG)hXJuD5WfOi4|W2`jXPx# diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py index 3cc7972aa..e63fe453e 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py @@ -1026,6 +1026,7 @@ def train_one_epoch( if batch_idx % params.accum_grads == 0: params.batch_idx_train += 1 batch_size = len(batch["supervisions"]["text"]) + print(batch_size) try: with torch.cuda.amp.autocast(enabled=params.use_fp16):