updated docs

This commit is contained in:
William Falcon
2019-09-17 09:53:31 -04:00
parent 4c61d1f30a
commit d3afc8acd5
+1 -1
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 from pytorch_lightning.callbacks import ModelCheckpoint
checkpoint_callback = ModelCheckpoint( checkpoint_callback = ModelCheckpoint(
filepath='/path/to/store/weights.ckpt', filepath='/path/to/store/weights/',
save_best_only=True, save_best_only=True,
verbose=True, verbose=True,
monitor='val_loss', monitor='val_loss',