Change num epochs in TF2 tutorial to 8
@@ -130,7 +130,7 @@ def main(_):
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)."
