fix: add score_function in saving best models (#145)

Author: ydcjeff

src/templates/template-vision-classification/utils.py

@@ -169,6 +169,8 @@ def setup_handlers(
         filename_prefix="best",
         n_saved=config.n_saved,
         global_step_transform=global_step_transform,
+        score_name="eval_accuracy",
+        score_function=Checkpoint.get_default_score_fn("eval_accuracy"),
     )
     evaluator.add_event_handler(
         Events.EPOCH_COMPLETED(every=1), ckpt_handler_eval

src/templates/template-vision-dcgan/trainers.py

@@ -158,7 +158,6 @@ def eval_function(engine: Engine, batch: Any):
         metrics = {
             "epoch": engine.state.epoch,
             "errD": errD.item(),
-            "eval_loss": errD.item(),
             "errG": errG.item(),
             "D_x": D_x,
             "D_G_z1": D_G_z1,

src/templates/template-vision-dcgan/utils.py

@@ -169,6 +169,8 @@ def setup_handlers(
         filename_prefix="best",
         n_saved=config.n_saved,
         global_step_transform=global_step_transform,
+        score_name="model_d_error",
+        score_function=Checkpoint.get_default_score_fn("errD", -1),
     )
     evaluator.add_event_handler(
         Events.EPOCH_COMPLETED(every=1), ckpt_handler_eval
@@ -179,7 +181,7 @@ def setup_handlers(
     #::: if (it.patience) { :::#
     # early stopping
     def score_fn(engine: Engine):
-        return -engine.state.metrics["eval_loss"]
+        return -engine.state.metrics["errD"]
 
     es = EarlyStopping(config.patience, score_fn, trainer)
     evaluator.add_event_handler(Events.EPOCH_COMPLETED, es)