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
  )