mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
2c8a06fde2
commit
96c802c46d
Binary file not shown.
@ -1027,6 +1027,7 @@ 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:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user