pytorch-lightning: Checkpoint gives error

Hi,

I wonder if we can only save the best model with lowest validation error, and not save the others checkpoints.

I took a look at checkpoint_callback’s save_best_only (below), it seems that this is for saving the best at every epoch (because the file name changes at every epoch). So I wonder if we can only save the best in the whole training process. Thanks! checkpoint_callback = ModelCheckpoint( filepath=os.getcwd(), save_best_only=True, verbose=True, monitor='val_loss', mode='min', prefix='' )

About this issue

  • Original URL
  • State: closed
  • Created 5 years ago
  • Comments: 19 (10 by maintainers)

Most upvoted comments

@Jiequannnnnnnnnn can you report it as a bug with reproducibility example…?

Sorry…I tried and not sure how to add a bug label. What does reproducibility example mean in this case?