diff --git a/src/attack/models.py b/src/attack/models.py index e494765..c1fe073 100644 --- 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, )