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:
@@ -12,7 +12,8 @@ checkpoint_callback = ModelCheckpoint(
|
|||||||
save_best_only=True,
|
save_best_only=True,
|
||||||
verbose=True,
|
verbose=True,
|
||||||
monitor='val_loss',
|
monitor='val_loss',
|
||||||
mode='min'
|
mode='min',
|
||||||
|
prefix=''
|
||||||
)
|
)
|
||||||
|
|
||||||
trainer = Trainer(checkpoint_callback=checkpoint_callback)
|
trainer = Trainer(checkpoint_callback=checkpoint_callback)
|
||||||
|
|||||||
Reference in New Issue
Block a user