from local

This commit is contained in:
dohe0342 2023-05-25 22:18:56 +09:00
parent f06b2a4792
commit 7154257f7a
2 changed files with 0 additions and 2 deletions

View File

@ -686,7 +686,6 @@ def main():
if '.pt' in params.model_name:
load_checkpoint(f"{params.exp_dir}/{params.model_name}", model)
'''
elif 'lora' in params.model_name:
load_checkpoint(f"{params.exp_dir}/../d2v-base-T.pt", model)
@ -703,7 +702,6 @@ def main():
lora.lora.load_state_dict(lora_param)
lora.lora.to(device)
logging.info("lora params load done")
'''
else:
if not params.use_averaged_model:
if params.iter > 0: