bug fixes

Nonbiuld · Oct 15, 2021 · f052294 · f052294
1 parent 72c1734
commit f052294
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchattacks/attacks/eotpgd.py b/torchattacks/attacks/eotpgd.py
@@ -75,7 +75,7 @@ def forward(self, images, labels):
                                             retain_graph=False,
                                             create_graph=False)[0]
 
-            adv_images = adv_images.detach() + self.alpha*grad.sign()
+            adv_images = adv_images.detach() + self.alpha*grad.sign()/self.eot_iter
             delta = torch.clamp(adv_images - images, min=-self.eps, max=self.eps)
             adv_images = torch.clamp(images + delta, min=0, max=1).detach()