from local

This commit is contained in:
dohe0342 2022-12-27 10:59:08 +09:00
parent bfbef65630
commit 7350e56647
2 changed files with 2 additions and 0 deletions

View File

@ -684,6 +684,8 @@ def load_checkpoint_if_available(
filename = params.exp_dir / f"checkpoint-{params.start_batch}.pt"
elif params.start_epoch > 1:
filename = params.exp_dir / f"epoch-{params.start_epoch-1}.pt"
elif params.add_adapter:
filename = params.exp_dir / f"d2v-base-T.pt"
else:
return None