facebookresearch/fairseq
1
1
Fork 0
mirror of https://github.com/facebookresearch/fairseq.git synced 2024-09-22 06:39:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix loading emissions (#1375)

Browse Source 
Summary:
broken in last change to infer.py

Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/1375

Reviewed By: xuqiantong

Differential Revision: D24531499

Pulled By: alexeib

fbshipit-source-id: fab60abf67a05c48e1ff750fac3ab6d4c0fa2770
This commit is contained in:
alexeib 2020-10-25 12:53:07 -07:00 committed by Facebook GitHub Bot
parent 6ee0364685
commit 3c41478083
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/speech_recognition/infer.py
View File

@ -269,6 +269,7 @@ def main(args, task=None, model_state=None):
# Load ensemble
if args.load_emissions:
models, criterions = [], []
task = tasks.setup_task(args)
else:
logger.info("| loading model(s) from {}".format(args.path))
models, criterions, task = load_models_and_criterions(
@ -282,6 +283,7 @@ def main(args, task=None, model_state=None):
# Load dataset splits
task.load_dataset(args.gen_subset)
# Set dictionary
tgt_dict = task.target_dictionary