diff --git a/ignite/handlers/checkpoint.py b/ignite/handlers/checkpoint.py index 6f4321b7812f..a1e4b441f6bc 100644 --- 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)