Update Universal.py

DSE-MSU · Apr 4, 2024 · 3155d9b · 3155d9b
1 parent 60bc64f
commit 3155d9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeprobust/image/attack/Universal.py b/deeprobust/image/attack/Universal.py
@@ -123,7 +123,7 @@ def universal_adversarial_perturbation(dataloader, model, device, xi=10, delta=0
     v.requires_grad_()
 
     fooling_rate = 0.0
-    num_images =  num_classes
+    num_images = len(dataloader)
     itr = 0
 
     while fooling_rate < 1-delta and itr < max_iter_uni: