generic-pretrained-GEC
generic-pretrained-GEC copied to clipboard
tasks/translation_from_pretrained_bart of mBART
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)