Merge pull request #5 from TLMichael/patch-2

a little bug fixed
MadryLab · Sep 16, 2019 · f624c78 · f624c78
2 parents 319791e + 1b3c345
commit f624c78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/robustness/train.py b/robustness/train.py
@@ -374,12 +374,12 @@ def _model_loop(args, loop_type, loader, model, opt, epoch, adv, writer):
     if adv:
         attack_kwargs = {
             'constraint': args.constraint,
-            'eps': args.eps,
+            'eps': eps,
             'step_size': args.attack_lr,
             'iterations': args.attack_steps,
             'random_start': False,
             'custom_loss': adv_criterion,
-            'random_restarts': args.random_restarts,
+            'random_restarts': random_restarts,
             'use_best': bool(args.use_best)
         }