removed checkpoint save_function option

This commit is contained in:
William Falcon
2019-06-28 17:14:18 -04:00
parent fd28d38693
commit 63d84283a4
3 changed files with 21 additions and 6 deletions
+18
View File
@@ -0,0 +1,18 @@
Lightning can automate saving and loading checkpoints.
---
### Model saving
To enable checkpointing, define the checkpoint callback
``` {.python}
from pytorch_lightning.utils.pt_callbacks import ModelCheckpoint
checkpoint = ModelCheckpoint(
filepath='/path/to/store/weights.ckpt',
save_function=None,
save_best_only=not hparams.keep_all_checkpoints,
verbose=True,
monitor=hparams.model_save_monitor_value,
mode=hparams.model_save_monitor_mode
)
```