from local

This commit is contained in:
dohe0342 2022-12-20 16:16:46 +09:00
parent 225e8b7beb
commit 6c82d84694
2 changed files with 77 additions and 74 deletions

View File

@ -660,6 +660,9 @@ 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_path:
load_checkpoint(f"{params.exp_dir}/epoch-{params.epoch}.pt", model)
else:
if not params.use_averaged_model: if not params.use_averaged_model:
if params.iter > 0: if params.iter > 0:
filenames = find_checkpoints(params.exp_dir, iteration=-params.iter)[ filenames = find_checkpoints(params.exp_dir, iteration=-params.iter)[