Remove set_return_type in lgv.py

mmajewsk · Jul 24, 2023 · 5dcecfc · 5dcecfc
1 parent c8b84e2
commit 5dcecfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchattacks/wrappers/lgv.py b/torchattacks/wrappers/lgv.py
@@ -161,7 +161,7 @@ def forward(self, images, labels):
             else:
                 raise NotImplementedError('Targeted attack mode not supported by LGV.')
         # set return type to base attack
-        self.base_attack.set_return_type(self.return_type)
+        # self.base_attack.set_return_type(self.return_type)
 
         adv_images = self.base_attack(images, labels)
         return adv_images