from local

This commit is contained in:
dohe0342 2022-12-20 16:17:07 +09:00
parent 559eb5e5de
commit 69bf6b5b0e
2 changed files with 1 additions and 1 deletions

View File

@ -661,7 +661,7 @@ def main():
model = get_transducer_model(params) model = get_transducer_model(params)
if params.model_name: if params.model_name:
load_checkpoint(f"{params.exp_dir}/epoch-{params.epoch}.pt", 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:
if params.iter > 0: if params.iter > 0: