mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
updated docs
This commit is contained in:
@@ -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',
|
||||||
|
|||||||
Reference in New Issue
Block a user