Skip to content

Commit 1b468e5

Browse files
authored
Update checkpoint.py (#1394)
1 parent 11f6d62 commit 1b468e5

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

ignite/handlers/checkpoint.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -220,9 +220,12 @@ def score_function(engine):
220220
return engine.state.metrics['accuracy']
221221
222222
to_save = {'model': model}
223-
handler = Checkpoint(to_save, DiskSaver('/tmp/models', create_dir=True), n_saved=2,
224-
filename_prefix='best', score_function=score_function, score_name="val_acc",
225-
global_step_transform=global_step_from_engine(trainer))
223+
handler = Checkpoint(
224+
to_save, DiskSaver('/tmp/models', create_dir=True),
225+
n_saved=2, filename_prefix='best',
226+
score_function=score_function, score_name="val_acc",
227+
global_step_transform=global_step_from_engine(trainer)
228+
)
226229
227230
evaluator.add_event_handler(Events.COMPLETED, handler)
228231

0 commit comments

Comments
 (0)