from local

This commit is contained in:
dohe0342 2023-04-18 14:00:43 +09:00
parent b0271a1987
commit d84ec859ff
2 changed files with 2 additions and 0 deletions

View File

@ -717,6 +717,8 @@ def load_checkpoint_if_available(
if "cur_batch_idx" in saved_params:
params["cur_batch_idx"] = saved_params["cur_batch_idx"]
params.batch_idx_train = 0
return saved_params