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)
Sorry…I tried and not sure how to add a bug label. What does reproducibility example mean in this case?