fairseq/modules/gumbel_vector_quantizer.py

Killed 0 out of 3 mutants

Survived

Survived mutation testing. These mutants show holes in your test suite.

Mutant 3372

--- fairseq/modules/gumbel_vector_quantizer.py
+++ fairseq/modules/gumbel_vector_quantizer.py
@@ -19,7 +19,7 @@
         vq_dim,
         time_first,
         activation=nn.GELU(),
-        weight_proj_depth=1,
+        weight_proj_depth=2,
         weight_proj_factor=1,
     ):
         """Vector quantization using gumbel softmax

Mutant 3373

--- fairseq/modules/gumbel_vector_quantizer.py
+++ fairseq/modules/gumbel_vector_quantizer.py
@@ -20,7 +20,7 @@
         time_first,
         activation=nn.GELU(),
         weight_proj_depth=1,
-        weight_proj_factor=1,
+        weight_proj_factor=2,
     ):
         """Vector quantization using gumbel softmax

Mutant 3374

--- fairseq/modules/gumbel_vector_quantizer.py
+++ fairseq/modules/gumbel_vector_quantizer.py
@@ -110,7 +110,7 @@
         res = self.forward(x, produce_targets=True)
         return res["x"], res["targets"]
 
-    def forward(self, x, produce_targets=False):
+    def forward(self, x, produce_targets=True):
 
         result = {"num_vars": self.num_vars * self.groups}