Update trades.py

yaodongyu · Sep 14, 2019 · c8233ad · c8233ad
1 parent c1cad08
commit c8233ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trades.py b/trades.py
@@ -65,7 +65,7 @@ def trades_loss(model,
             # projection
             delta.data.add_(x_natural)
             delta.data.clamp_(0, 1).sub_(x_natural)
-            delta.data.renorm_(p=2, dim=0, maxnorm=max_norm)
+            delta.data.renorm_(p=2, dim=0, maxnorm=epsilon)
         x_adv = Variable(x_natural + delta, requires_grad=False)
     else:
         x_adv = torch.clamp(x_adv, 0.0, 1.0)