Merge pull request #15 from simone-dotolo:main

DIFGSM fix input_diversity transform probability
spencerwooo · Nov 14, 2024 · a49ae05 · a49ae05
2 parents 5ab4ce9 + c1c6e28
commit a49ae05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torchattack/difgsm.py b/src/torchattack/difgsm.py
@@ -134,7 +134,7 @@ def input_diversity(
         The diversified batch of images. Shape: (N, C, H, W).
     """
 
-    if torch.rand(1) < diversity_prob:
+    if torch.rand(1) > diversity_prob:
         return x
 
     img_size = x.shape[-1]