next

stargan/attacks.py

@@ -7,7 +7,7 @@ import torch
 import torch.nn as nn
 
 class LinfPGDAttack(object):
-    def __init__(self, model=None, device=None, epsilon=0.05, k=1, a=0.05, feat = None):
+    def __init__(self, model=None, device=None, epsilon=0.05, k=1, a=0.01, feat = None):
         self.model = model
         self.epsilon = epsilon
         self.k = k
@@ -24,7 +24,7 @@ class LinfPGDAttack(object):
         X = X_nat.clone().detach_()
 
         for i in range(self.k):
-            # print(i)
+            print(i)
             X.requires_grad = True
             output, feats = self.model(X, c_trg)