next

cyclegan/test.py

@@ -70,9 +70,9 @@ if __name__ == '__main__':
         model.set_input(data)  # unpack data from data loader
         with torch.no_grad():
             model.forward_noattack()
-        # if i == 0:
-        #     input_adv, perturb = model.attack()
-        input_adv, perturb = model.attack()
+        if i == 0:
+            input_adv, perturb = model.attack()
+        # input_adv, perturb = model.attack()
         with torch.no_grad():
             model.forward_attack(perturb)
             model.compute_visuals()

cyclegan/util/attacks.py

@@ -7,7 +7,7 @@ import torch
 import torch.nn as nn
 
 class LinfPGDAttack(object):
-    def __init__(self, model=None, epsilon=0.05, k=20, a=0.01):
+    def __init__(self, model=None, epsilon=0.05, k=1, a=0.05):
         self.model = model
         self.epsilon = epsilon
         self.k = k