Change num epochs in CIFAR-10 TF2 tutorial to 8

cleverhans-lab · Mar 16, 2021 · e3ea561 · e3ea561
1 parent 7fcfa20
commit e3ea561
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/tf2/cifar10_tutorial.py b/tutorials/tf2/cifar10_tutorial.py
@@ -130,7 +130,7 @@ def train_step(x, y):
 
 
 if __name__ == "__main__":
-    flags.DEFINE_integer("nb_epochs", 200, "Number of epochs.")
+    flags.DEFINE_integer("nb_epochs", 8, "Number of epochs.")
     flags.DEFINE_float("eps", 0.05, "Total epsilon for FGM and PGD attacks.")
     flags.DEFINE_bool(
         "adv_train", False, "Use adversarial training (on PGD adversarial examples)."