Fix generative.py in generative_loss

fcdl94 · Mar 10, 2021 · 155d07f · 155d07f
1 parent 5a0d665
commit 155d07f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/methods/generative.py b/methods/generative.py
@@ -142,7 +142,7 @@ def generative_loss(self, images=None, labels=None):
 
     def generate_synth_feat(self, images=None, labels=None):
         real_feat, real_lbl = self.get_real_features(self.model, images, labels)
-        masked_feat, masked_lbl, real_feat, real_lbl = self.mask_func(real_feat, real_lbl)
+        masked_feat, masked_lbl, real_feat, real_lbl, noise = self.mask_func(real_feat, real_lbl)
 
         gen_feat = self.generator(masked_feat, add_z=self.add_Z)