temp

Author: leej3

ignite/engine/engine.py

@@ -129,7 +129,16 @@ def compute_mean_std(engine, batch):
     """
 
     _state_dict_all_req_keys = ("epoch_length", "max_epochs")
-    _state_dict_one_of_opt_keys = ("iteration", "epoch")
+    _state_dict_one_of_opt_keys = (
+        (
+            "iteration",
+            "epoch",
+        ),
+        (
+            "max_epochs",
+            "max_iters",
+        ),
+    )
 
     # Flag to disable engine._internal_run as generator feature for BC
     interrupt_resume_enabled = True