from local

This commit is contained in:
dohe0342 2023-05-24 21:16:56 +09:00
parent b972cbaff2
commit 9b45d8bd93
2 changed files with 2 additions and 2 deletions

View File

@ -685,8 +685,8 @@ def main():
if '.pt' in params.model_name:
load_checkpoint(f"{params.exp_dir}/{params.model_name}", model)
elif 'lora' in params.model_name:
elif 'lora' in params.model_name:
load_checkpoint(f"{params.exp_dir}/../d2v-base-T.pt", model)
else:
if not params.use_averaged_model:
if params.iter > 0: