mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-18 21:44:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Also set epoch

Browse Source
This commit is contained in:
Nickolay Shmyrev 2023-07-03 18:20:16 +02:00
parent d19f4bb009
commit a4f6069029
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
icefall/rnn_lm/train.py
View File

@ -278,7 +278,7 @@ def load_checkpoint_if_available(
elif params.start_epoch > 1: elif params.start_epoch > 1:
filename = params.exp_dir / f"epoch-{params.start_epoch-1}.pt" filename = params.exp_dir / f"epoch-{params.start_epoch-1}.pt"
else: else:
return return None
logging.info(f"Loading checkpoint: {filename}") logging.info(f"Loading checkpoint: {filename}")
saved_params = load_checkpoint( saved_params = load_checkpoint(
@ -298,6 +298,10 @@ def load_checkpoint_if_available(
for k in keys: for k in keys:
params[k] = saved_params[k] params[k] = saved_params[k]
if params.start_batch > 0:
if "cur_epoch" in saved_params:
params["start_epoch"] = saved_params["cur_epoch"]
return saved_params return saved_params