Merge pull request cleverhans-lab#1206 from grahamannett/master

cast to tensor if in numpy array for fgsm attack
riccitensor · Mar 31, 2021 · cb5a1bf · cb5a1bf
2 parents 1233a4a + e719317
commit cb5a1bf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cleverhans/tf2/attacks/fast_gradient_method.py b/cleverhans/tf2/attacks/fast_gradient_method.py
@@ -55,6 +55,9 @@ def fast_gradient_method(
     if clip_max is not None:
         asserts.append(tf.math.less_equal(x, clip_max))
 
+    # cast to tensor if provided as numpy array
+    x = tf.cast(x, tf.float32)
+
     if y is None:
         # Using model predictions as ground truth to avoid label leaking
         y = tf.argmax(model_fn(x), 1)