mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-03-30 00:01:48 +09:00
parent 6b201624b9
commit f5ed700ada
3 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/.run_adapter.sh.swp
View File

Binary file not shown.

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp
View File

Binary file not shown.

2
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py
View File

@ -1027,7 +1027,6 @@ def train_one_epoch(
scheduler_enc, scheduler_dec = scheduler[0], scheduler[1] scheduler_enc, scheduler_dec = scheduler[0], scheduler[1]
for batch_idx, batch in enumerate(train_dl): 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: if params.batch_idx_train > params.num_updates:
break break
if batch_idx < cur_batch_idx: 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) scaler = GradScaler(enabled=params.use_fp16, init_scale=1.0)
for epoch in range(params.start_epoch, params.num_epochs + 1): 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) scheduler.step_epoch(epoch - 1)
fix_random_seed(params.seed + epoch - 1) fix_random_seed(params.seed + epoch - 1)
train_dl.sampler.set_epoch(epoch - 1) train_dl.sampler.set_epoch(epoch - 1)