from local

This commit is contained in:
dohe0342 2022-12-20 16:16:55 +09:00
parent 6c82d84694
commit 559eb5e5de
2 changed files with 1 additions and 1 deletions

View File

@ -660,7 +660,7 @@ def main():
logging.info("About to create model")
model = get_transducer_model(params)
if params.model_path:
if params.model_name:
load_checkpoint(f"{params.exp_dir}/epoch-{params.epoch}.pt", model)
else:
if not params.use_averaged_model: