(#7) Attack: Cast input to float32 and fix params

betarixm · Apr 18, 2022 · 541490c · 541490c
1 parent 220064a
commit 541490c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/attack/models.py b/src/attack/models.py
@@ -17,5 +17,10 @@ def add_perturbation(self, x: np.array) -> np.array:
 class Pgd(Attack):
     def add_perturbation(self, x: np.array) -> np.array:
         return projected_gradient_descent(
-            self.victim_model.model(), x, 0.05, 0.01, 40, np.inf
+            self.victim_model.model(),
+            tf.cast(x, tf.float32),
+            8 / 255,
+            1 / 255,
+            10,
+            np.inf,
         )