@@ -77,7 +77,7 @@ class TestModel(BaseModel):
        pgd_attack = attacks.LinfPGDAttack(model=self.netG)
        black = np.zeros((1, 3, image.size(2), image.size(3)))
        black = torch.FloatTensor(black).cuda()
-        input_adv, perturb = pgd_attack.perturb(image, black)      
+        input_adv, perturb = pgd_attack.perturb(image, image)      
        
        return input_adv, perturb

@@ -27,7 +27,7 @@ class LinfPGDAttack(object):
            output = self.model(X)

            self.model.zero_grad()
-            loss = -self.loss_fn(output, y)
+            loss = self.loss_fn(output, y)
            loss.backward()
            grad = X.grad