from local

This commit is contained in:
dohe0342 2023-04-27 15:31:03 +09:00
parent 5b4cd838b1
commit 36ac0ceaab
2 changed files with 1 additions and 1 deletions

View File

@ -724,7 +724,7 @@ def load_checkpoint_if_available(
if "cur_batch_idx" in saved_params: if "cur_batch_idx" in saved_params:
params["cur_batch_idx"] = saved_params["cur_batch_idx"] params["cur_batch_idx"] = saved_params["cur_batch_idx"]
params.batch_idx_train = 0 #params.batch_idx_train = 0
return saved_params return saved_params