From c1c6e2827d167fa68a0f8447acd31c3966d8e5af Mon Sep 17 00:00:00 2001
From: simone-dotolo <simonedotolo@libero.it>
Date: Mon, 11 Nov 2024 14:51:02 +0100
Subject: [PATCH] input_diversity probability

---
 src/torchattack/difgsm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/torchattack/difgsm.py b/src/torchattack/difgsm.py
index 53e6d08..b66d5e2 100644
--- 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]