cast to tensor if in numpy array for fgsm attack

cleverhans-lab · Mar 30, 2021 · e719317 · e719317
1 parent 1233a4a
commit e719317
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)