Fix format issue.
@@ -495,12 +495,14 @@ class SaliencyMapMethod(Attack):
return result
labels, nb_classes = self.get_or_guess_labels(x, kwargs)
- self.y_target = tf.py_func(random_targets, [labels], tf.float32)
+ self.y_target = tf.py_func(random_targets, [labels],
+ tf.float32)
self.y_target.set_shape([None, nb_classes])
x_adv = jsma_symbolic(x, model=self.model, y_target=self.y_target,
theta=self.theta, gamma=self.gamma,
- clip_min=self.clip_min, clip_max=self.clip_max)
+ clip_min=self.clip_min,
+ clip_max=self.clip_max)
else:
from .attacks_tf import jacobian_graph, jsma_batch
