from local

This commit is contained in:
dohe0342 2023-03-29 22:20:42 +09:00
parent 9561e30284
commit 8f5af47c26
3 changed files with 2 additions and 0 deletions

View File

@ -739,6 +739,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