wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #347 (#393)

Browse Source
This commit is contained in:
William Falcon
2019-10-19 00:51:48 +02:00
committed by GitHub
parent b0281395bf
commit 58d52c25a1
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/Trainer/Training Loop.md
+9 -2
View File
@@ -35,15 +35,22 @@ early_stop_callback = EarlyStopping(
mode='min' mode='min'
) )
# without passing anything in, uses the default callback above
trainer = Trainer()
# pass in your own to override the default callback
trainer = Trainer(early_stop_callback=early_stop_callback) trainer = Trainer(early_stop_callback=early_stop_callback)
# pass in None to disable it
trainer = Trainer(early_stop_callback=None)
``` ```
--- ---
#### Force disable early stop #### Force disable early stop
Use this to turn off early stopping and run training to the [max_epoch](#force-training-for-min-or-max-epochs) To disable early stopping pass None to the early_stop_callback
``` {.python} ``` {.python}
# DEFAULT # DEFAULT
trainer = Trainer(enable_early_stop=True) trainer = Trainer(early_stop_callback=None)
``` ```
--- ---