facebookresearch/fairseq
1
1
Fork 0
mirror of https://github.com/facebookresearch/fairseq.git synced 2024-08-16 12:00:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

query tgt_dict after loading task_state (#2019)

Browse Source 
Summary:
# Before submitting
`self.task.target_dictionary` is queried before `task_state` is loaded (in `self.load_model_ensemble()`).

## What does this PR do?
Fix the bug above

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

Reviewed By: alexeib

Differential Revision: D29523921

Pulled By: wnhsu

fbshipit-source-id: 763b504dc1b4899e623eaa5c19972cec9d0a8985
This commit is contained in:
Wei-Ning Hsu 2021-07-01 13:11:40 -07:00 committed by Facebook GitHub Bot
parent 096f492a22
commit cdc1a553eb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/speech_recognition/new/infer.py
View File

@ -99,11 +99,11 @@ class InferenceProcessor:
def __init__(self, cfg: InferConfig) -> None:
self.cfg = cfg
self.task = tasks.setup_task(cfg.task)
self.tgt_dict = self.task.target_dictionary
models, saved_cfg = self.load_model_ensemble()
self.models = models
self.saved_cfg = saved_cfg
self.tgt_dict = self.task.target_dictionary
self.task.load_dataset(
self.cfg.dataset.gen_subset,