Merge branch 'master' into sphinx-autosummary

pytorch · Oct 20, 2020 · 87f3c42 · 87f3c42
2 parents d9c73da + 1b468e5
commit 87f3c42
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/ignite/handlers/checkpoint.py b/ignite/handlers/checkpoint.py
@@ -220,9 +220,12 @@ def score_function(engine):
                 return engine.state.metrics['accuracy']
 
             to_save = {'model': model}
-            handler = Checkpoint(to_save, DiskSaver('/tmp/models', create_dir=True), n_saved=2,
-                                 filename_prefix='best', score_function=score_function, score_name="val_acc",
-                                 global_step_transform=global_step_from_engine(trainer))
+            handler = Checkpoint(
+                to_save, DiskSaver('/tmp/models', create_dir=True),
+                n_saved=2, filename_prefix='best',
+                score_function=score_function, score_name="val_acc",
+                global_step_transform=global_step_from_engine(trainer)
+            )
 
             evaluator.add_event_handler(Events.COMPLETED, handler)