mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-26 13:00:58 +08:00
updated docs
This commit is contained in:
1 parent
4c61d1f30a
commit
d3afc8acd5
1 file changed
+1
-1
@@ -8,7 +8,7 @@ To enable checkpointing, define the checkpoint callback and give it to the train
|
||||
from pytorch_lightning.callbacks import ModelCheckpoint
|
||||
|
||||
checkpoint_callback = ModelCheckpoint(
|
||||
filepath='/path/to/store/weights.ckpt',
|
||||
filepath='/path/to/store/weights/',
|
||||
save_best_only=True,
|
||||
verbose=True,
|
||||
monitor='val_loss',
|
||||
|
||||
Reference in new issue
Block a user