fairseq/tasks/translation_from_pretrained_bart.py

Survived

Mutant 561

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -11,7 +11,7 @@
 from . import register_task
 
 
-@register_task('translation_from_pretrained_bart')
+@register_task('XXtranslation_from_pretrained_bartXX')
 class TranslationFromPretrainedBARTTask(TranslationTask):
     """
     Translate from source language to target language with a model initialized with a multilingual pretrain.

Mutant 562

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -10,8 +10,6 @@
 from .translation import load_langpair_dataset, TranslationTask
 from . import register_task
 
-
-@register_task('translation_from_pretrained_bart')
 class TranslationFromPretrainedBARTTask(TranslationTask):
     """
     Translate from source language to target language with a model initialized with a multilingual pretrain.

Mutant 563

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -33,7 +33,6 @@
         :prog:
     """
 
-    @staticmethod
     def add_args(parser):
         """Add task-specific arguments to the parser."""
         # fmt: off

Mutant 565

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -38,7 +38,7 @@
         """Add task-specific arguments to the parser."""
         # fmt: off
         TranslationTask.add_args(parser)
-        parser.add_argument('--langs', required=True, metavar='LANG',
+        parser.add_argument('--langs', required=False, metavar='LANG',
                             help='comma-separated list of monolingual language, '
                                  'for example, "en,de,fr". These should match the '
                                  'langs from pretraining (and be in the same order). '

Mutant 566

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -38,7 +38,7 @@
         """Add task-specific arguments to the parser."""
         # fmt: off
         TranslationTask.add_args(parser)
-        parser.add_argument('--langs', required=True, metavar='LANG',
+        parser.add_argument('--langs', required=True, metavar='XXLANGXX',
                             help='comma-separated list of monolingual language, '
                                  'for example, "en,de,fr". These should match the '
                                  'langs from pretraining (and be in the same order). '

Mutant 567

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -39,7 +39,7 @@
         # fmt: off
         TranslationTask.add_args(parser)
         parser.add_argument('--langs', required=True, metavar='LANG',
-                            help='comma-separated list of monolingual language, '
+                            help='XXcomma-separated list of monolingual language, XX'
                                  'for example, "en,de,fr". These should match the '
                                  'langs from pretraining (and be in the same order). '
                                  'You should always add all pretraining language idx '

Mutant 568

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -44,7 +44,7 @@
                                  'langs from pretraining (and be in the same order). '
                                  'You should always add all pretraining language idx '
                                  'during finetuning.')
-        parser.add_argument('--prepend-bos', action='store_true',
+        parser.add_argument('XX--prepend-bosXX', action='store_true',
                             help='prepend bos token to each sentence, which matches '
                                  'mBART pretraining')
         # fmt: on

Mutant 570

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -45,7 +45,7 @@
                                  'You should always add all pretraining language idx '
                                  'during finetuning.')
         parser.add_argument('--prepend-bos', action='store_true',
-                            help='prepend bos token to each sentence, which matches '
+                            help='XXprepend bos token to each sentence, which matches XX'
                                  'mBART pretraining')
         # fmt: on
 

Mutant 571

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -57,7 +57,7 @@
                 d.add_symbol('[{}]'.format(l))
             d.add_symbol('')
 
-    def load_dataset(self, split, epoch=1, combine=False, **kwargs):
+    def load_dataset(self, split, epoch=2, combine=False, **kwargs):
         """Load a given dataset split.
 
         Args:

Mutant 572

--- fairseq/tasks/translation_from_pretrained_bart.py
+++ fairseq/tasks/translation_from_pretrained_bart.py
@@ -57,7 +57,7 @@
                 d.add_symbol('[{}]'.format(l))
             d.add_symbol('')
 
-    def load_dataset(self, split, epoch=1, combine=False, **kwargs):
+    def load_dataset(self, split, epoch=1, combine=True, **kwargs):
         """Load a given dataset split.
 
         Args: