updated docs

This commit is contained in:
William Falcon 2019-09-17 09:53:31 -04:00
parent 4c61d1f30a
commit d3afc8acd5
1 changed files with 1 additions and 1 deletions

View File

@ -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',