generic-pretrained-GEC icon indicating copy to clipboard operation
generic-pretrained-GEC copied to clipboard

tasks/translation_from_pretrained_bart of mBART

Open AutoTemp opened this issue 3 years ago • 0 comments

I think line 118 of tasks/translation_from_pretrained_bart.py may need to be revised. dataset = LanguagePairDataset(src_tokens, src_lengths, self.source_dictionary) --> dataset = LanguagePairDataset(source_tokens, src_lengths, self.source_dictionary)

AutoTemp avatar Apr 13 '21 08:04 AutoTemp