fairseq/optim/nag.py

Survived

Mutant 2916

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -9,7 +9,7 @@
 from . import FairseqOptimizer, register_optimizer
 
 
-@register_optimizer('nag')
+@register_optimizer('XXnagXX')
 class FairseqNAG(FairseqOptimizer):
     def __init__(self, args, params):
         super().__init__(args)

Mutant 2917

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -8,8 +8,6 @@
 
 from . import FairseqOptimizer, register_optimizer
 
-
-@register_optimizer('nag')
 class FairseqNAG(FairseqOptimizer):
     def __init__(self, args, params):
         super().__init__(args)

Mutant 2918

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -15,7 +15,6 @@
         super().__init__(args)
         self._optimizer = NAG(params, **self.optimizer_config)
 
-    @staticmethod
     def add_args(parser):
         """Add optimizer-specific arguments to the parser."""
         # fmt: off

Mutant 2919

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -23,9 +23,7 @@
                             help='momentum factor')
         parser.add_argument('--weight-decay', '--wd', default=0.0, type=float, metavar='WD',
                             help='weight decay')
-        # fmt: on
 
-    @property
     def optimizer_config(self):
         """
         Return a kwarg dictionary that will be used to override optimizer

Mutant 2920

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -41,7 +41,7 @@
 
 
 class NAG(Optimizer):
-    def __init__(self, params, lr=required, momentum=0, weight_decay=0):
+    def __init__(self, params, lr=required, momentum=1, weight_decay=0):
         defaults = dict(lr=lr, lr_old=lr, momentum=momentum, weight_decay=weight_decay)
         super(NAG, self).__init__(params, defaults)
 

Mutant 2921

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -41,7 +41,7 @@
 
 
 class NAG(Optimizer):
-    def __init__(self, params, lr=required, momentum=0, weight_decay=0):
+    def __init__(self, params, lr=required, momentum=0, weight_decay=1):
         defaults = dict(lr=lr, lr_old=lr, momentum=momentum, weight_decay=weight_decay)
         super(NAG, self).__init__(params, defaults)
 

Mutant 2922

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -45,7 +45,6 @@
         defaults = dict(lr=lr, lr_old=lr, momentum=momentum, weight_decay=weight_decay)
         super(NAG, self).__init__(params, defaults)
 
-    @property
     def supports_memory_efficient_fp16(self):
         return True
 

Mutant 2923

--- fairseq/optim/nag.py
+++ fairseq/optim/nag.py
@@ -49,7 +49,6 @@
     def supports_memory_efficient_fp16(self):
         return True
 
-    @property
     def supports_flat_params(self):
         return True