I use the ModelCheckpoint handler in Ignite to save the model using these lines
ckpt_handler = ModelCheckpoint(f'./experiments/{log_num}/checkpoints', filename_prefix='epoch_', n_saved=None, create_dir=True)
trainer.add_event_handler(Events.EPOCH_COMPLETED(every=2), ckpt_handler, to_save)

but the .pth filename follow this pattern --> epoch__checkpoint_344.pth (344 refers to iterations).
I would to include the number of epoch into the filename… Any idea how’s that done?