mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-26 13:00:58 +08:00
updated docs
This commit is contained in:
1 parent
d3afc8acd5
commit
2a1bc22f42
1 file changed
+2
-1
@@ -12,7 +12,8 @@ checkpoint_callback = ModelCheckpoint(
|
||||
save_best_only=True,
|
||||
verbose=True,
|
||||
monitor='val_loss',
|
||||
mode='min'
|
||||
mode='min',
|
||||
prefix=''
|
||||
)
|
||||
|
||||
trainer = Trainer(checkpoint_callback=checkpoint_callback)
|
||||
|
||||
Reference in new issue
Block a user