diff --git a/fluid/adversarial/advbox/attacks/gradient_method.py b/fluid/adversarial/advbox/attacks/gradient_method.py index 47c1ddffbbd289b43fbfd7f323bc3f5b53e58b0a..e7d42c4f95a126e6566dc8f3949038358481c948 100644 --- a/fluid/adversarial/advbox/attacks/gradient_method.py +++ b/fluid/adversarial/advbox/attacks/gradient_method.py @@ -32,7 +32,7 @@ class GradientMethodAttack(Attack): super(GradientMethodAttack, self).__init__(model) self.support_targeted = support_targeted - def _apply(self, adversary, norm_ord=2, epsilons=0.01, steps=100): + def _apply(self, adversary, norm_ord=np.inf, epsilons=0.01, steps=100): """ Apply the gradient attack method. :param adversary(Adversary):