fix: bump max_epochs to 5, get_handlers arguments (#58)

pytorch-ignite · Apr 10, 2021 · 818585e · 818585e
1 parent e79bff3
commit 818585e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/templates/gan/_sidebar.py b/templates/gan/_sidebar.py
@@ -43,7 +43,7 @@ def optimizer_options(config):
 
 def training_options(config):
     st.markdown("## Training Options")
-    config["max_epochs"] = st.number_input("Maximum epochs to train (max_epochs)", min_value=1, value=2)
+    config["max_epochs"] = st.number_input("Maximum epochs to train (max_epochs)", min_value=1, value=5)
     st.markdown("---")
 
 

diff --git a/templates/gan/main.py b/templates/gan/main.py
@@ -90,6 +90,9 @@ def run(local_rank: int, config: Any, *args: Any, **kwargs: Any):
         config=config,
         model={'netD', netD, 'netG', netG},
         train_engine=train_engine,
+        eval_engine=None,
+        metric_name=None,
+        es_metric_name=None,
         to_save=to_save,
         lr_scheduler=lr_scheduler,
         output_names=["errD", "errG", "D_x", "D_G_z1", "D_G_z2"],

diff --git a/templates/image_classification/_sidebar.py b/templates/image_classification/_sidebar.py
@@ -51,7 +51,7 @@ def optimizer_options(config):
 
 def training_options(config):
     st.markdown("## Training Options")
-    config["max_epochs"] = st.number_input("Maximum epochs to train (max_epochs)", min_value=1, value=2)
+    config["max_epochs"] = st.number_input("Maximum epochs to train (max_epochs)", min_value=1, value=5)
     config["num_warmup_epochs"] = st.number_input(
         "number of warm-up epochs before learning rate decay (num_warmup_epochs)", min_value=1, value=4
     )