Update fgsm.py

tomcarter23 · Dec 4, 2024 · f64deda · f64deda
1 parent 787b4b0
commit f64deda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/adversarial_attack/fgsm.py b/adversarial_attack/fgsm.py
@@ -126,7 +126,7 @@ def targeted_attack(
         grad = compute_gradient(model=model, input=adv_tensor, target=target)
         adv_tensor = torch.clamp(adv_tensor - epsilon * grad.sign(), -2, 2)
         new_pred = model(adv_tensor).argmax()
-        if orig_pred_idx.item() != new_pred:
+        if orig_pred_idx.item() == target.item():
             return adv_tensor, orig_pred, new_pred
 
     warnings.warn(