fairseq/tasks/translation_lev.py

Survived

Mutant 2046

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -12,7 +12,7 @@
 from fairseq.tasks.translation import TranslationTask, load_langpair_dataset
 from fairseq import utils
 
-@register_task('translation_lev')
+@register_task('XXtranslation_levXX')
 class TranslationLevenshteinTask(TranslationTask):
     """
     Translation (Sequence Generation) task for Levenshtein Transformer

Mutant 2047

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -12,7 +12,6 @@
 from fairseq.tasks.translation import TranslationTask, load_langpair_dataset
 from fairseq import utils
 
-@register_task('translation_lev')
 class TranslationLevenshteinTask(TranslationTask):
     """
     Translation (Sequence Generation) task for Levenshtein Transformer

Mutant 2048

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -19,7 +19,6 @@
     See `"Levenshtein Transformer" `_.
     """
 
-    @staticmethod
     def add_args(parser):
         """Add task-specific arguments to the parser."""
         # fmt: off

Mutant 2049

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -25,7 +25,7 @@
         # fmt: off
         TranslationTask.add_args(parser)
         parser.add_argument(
-            '--noise',
+            'XX--noiseXX',
             default='random_delete',
             choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
 

Mutant 2050

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -26,7 +26,7 @@
         TranslationTask.add_args(parser)
         parser.add_argument(
             '--noise',
-            default='random_delete',
+            default='XXrandom_deleteXX',
             choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
 
     def load_dataset(self, split, epoch=1, combine=False, **kwargs):

Mutant 2051

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -27,7 +27,7 @@
         parser.add_argument(
             '--noise',
             default='random_delete',
-            choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
+            choices=['XXrandom_deleteXX', 'random_mask', 'no_noise', 'full_mask'])
 
     def load_dataset(self, split, epoch=1, combine=False, **kwargs):
         """Load a given dataset split.

Mutant 2052

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -27,7 +27,7 @@
         parser.add_argument(
             '--noise',
             default='random_delete',
-            choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
+            choices=['random_delete', 'XXrandom_maskXX', 'no_noise', 'full_mask'])
 
     def load_dataset(self, split, epoch=1, combine=False, **kwargs):
         """Load a given dataset split.

Mutant 2053

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -27,7 +27,7 @@
         parser.add_argument(
             '--noise',
             default='random_delete',
-            choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
+            choices=['random_delete', 'random_mask', 'XXno_noiseXX', 'full_mask'])
 
     def load_dataset(self, split, epoch=1, combine=False, **kwargs):
         """Load a given dataset split.

Mutant 2054

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -27,7 +27,7 @@
         parser.add_argument(
             '--noise',
             default='random_delete',
-            choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
+            choices=['random_delete', 'random_mask', 'no_noise', 'XXfull_maskXX'])
 
     def load_dataset(self, split, epoch=1, combine=False, **kwargs):
         """Load a given dataset split.

Mutant 2055

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -29,7 +29,7 @@
             default='random_delete',
             choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
 
-    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 2056

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -29,7 +29,7 @@
             default='random_delete',
             choices=['random_delete', 'random_mask', 'no_noise', 'full_mask'])
 
-    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:

Mutant 2057

--- fairseq/tasks/translation_lev.py
+++ fairseq/tasks/translation_lev.py
@@ -145,7 +145,7 @@
                    criterion,
                    optimizer,
                    update_num,
-                   ignore_grad=False):
+                   ignore_grad=True):
         model.train()
         sample['prev_target'] = self.inject_noise(sample['target'])
         loss, sample_size, logging_output = criterion(model, sample)