Commit 817fccf5 authored by Maksym Del's avatar Maksym Del Committed by Facebook Github Bot
Browse files

Fix semisupervised_translation task (#706)

Summary:
Pass required "sample_key" argument to forward-backward call in semi-supervised task.
Pull Request resolved: https://github.com/pytorch/fairseq/pull/706

Differential Revision: D15217957

Pulled By: pipibjc

fbshipit-source-id: bf943d566c5caa67682dfb16ff8b7c432323cdba
parent 39cd4ce2
...@@ -345,7 +345,7 @@ class SemisupervisedTranslationTask(MultilingualTranslationTask): ...@@ -345,7 +345,7 @@ class SemisupervisedTranslationTask(MultilingualTranslationTask):
if self.lambda_parallel > 0.0: if self.lambda_parallel > 0.0:
for lang_pair in self.args.lang_pairs: for lang_pair in self.args.lang_pairs:
forward_backward(model.models[lang_pair], sample[lang_pair], self.lambda_parallel) forward_backward(model.models[lang_pair], sample[lang_pair], lang_pair, self.lambda_parallel)
if self.lambda_otf_bt > 0.0: if self.lambda_otf_bt > 0.0:
for lang_pair in self.args.lang_pairs: for lang_pair in self.args.lang_pairs:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment