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

Author: Jeff Yang

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("---")
 
 

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"],

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
     )