From f5ed700adab88f177509ccf178284612d3f4791c Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 30 Mar 2023 00:01:48 +0900 Subject: [PATCH] from local --- egs/librispeech/ASR/.run_adapter.sh.swp | Bin 12288 -> 12288 bytes .../.train_adapter.py.swp | Bin 94208 -> 94208 bytes .../train_adapter.py | 2 +- 3 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/.run_adapter.sh.swp b/egs/librispeech/ASR/.run_adapter.sh.swp index 9830ca2b284cedc082a5f90d77edace3538a19d9..799dbe4a6a4c94766e17fd84d115e339a905743b 100644 GIT binary patch delta 32 mcmZojXh;xGG6?hZRj|}EU;qLE1_r4hm6WTuOE!wX)CT~E@d*_G delta 32 mcmZojXh;xGG6?hZRj|}EU;qLE28PKtDk*>Z3O9=?2~VK@@zivT!MM>gTpM7 zvv%@MZ%}76np}9qVYAHZ7bd(qK$j-~^?+RNwp~zxQJj&r2*^*^uB*ss%r;$7ozZXm JI(5b%4*>2GHQ4|F delta 252 zcmZp8z}oPDb%TK*qx)ta$Hg?3Gx~V|00AI4WB>pF 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 529d9b9f7..42c7c0137 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 @@ -1027,7 +1027,6 @@ def train_one_epoch( scheduler_enc, scheduler_dec = scheduler[0], scheduler[1] for batch_idx, batch in enumerate(train_dl): - logging.info(f"update num : {params.batch_idx_train}") if params.batch_idx_train > params.num_updates: break if batch_idx < cur_batch_idx: @@ -1631,6 +1630,7 @@ def run_adapter(rank, world_size, args, wb=None): scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0) for epoch in range(params.start_epoch, params.num_epochs + 1): + logging.info(f"update num : {params.batch_idx_train}") scheduler.step_epoch(epoch - 1) fix_random_seed(params.seed + epoch - 1) train_dl.sampler.set_epoch(epoch - 1)