(#7) Attack: Fix CW Attack params

betarixm · Apr 21, 2022 · 4d13337 · 4d13337
1 parent 75f2e6a
commit 4d13337
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/attack/models.py b/src/attack/models.py
@@ -32,11 +32,11 @@ def add_perturbation(self, x: np.array) -> np.array:
         return carlini_wagner_l2(
             self.victim_model.model(),
             tf.cast(x, tf.float32),
-            batch_size=9,
+            batch_size=32,
             clip_min=0.0,
             clip_max=1.0,
-            binary_search_steps=9,
+            binary_search_steps=16,
             max_iterations=1000,
-            initial_const=10,
-            learning_rate=1e-2,
+            initial_const=8,
+            learning_rate=0.05,
         )