updated docs
This commit is contained in:
parent
4c61d1f30a
commit
d3afc8acd5
|
@ -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',
|
||||
|
|
Loading…
Reference in New Issue