from local

This commit is contained in:
dohe0342 2023-05-26 11:51:18 +09:00
parent 05b681d7d2
commit 45dbc89145
2 changed files with 1 additions and 1 deletions

View File

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