Skip to content

Commit 58d52c2

Browse files
Fixes #347 (#393)
1 parent b028139 commit 58d52c2

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

docs/Trainer/Training Loop.md

+9-2
Original file line numberDiff line numberDiff line change
@@ -35,15 +35,22 @@ early_stop_callback = EarlyStopping(
3535
mode='min'
3636
)
3737
38+
# without passing anything in, uses the default callback above
39+
trainer = Trainer()
40+
41+
# pass in your own to override the default callback
3842
trainer = Trainer(early_stop_callback=early_stop_callback)
43+
44+
# pass in None to disable it
45+
trainer = Trainer(early_stop_callback=None)
3946
```
4047

4148
---
4249
#### Force disable early stop
43-
Use this to turn off early stopping and run training to the [max_epoch](#force-training-for-min-or-max-epochs)
50+
To disable early stopping pass None to the early_stop_callback
4451
``` {.python}
4552
# DEFAULT
46-
trainer = Trainer(enable_early_stop=True)
53+
trainer = Trainer(early_stop_callback=None)
4754
```
4855

4956
---

0 commit comments

Comments
 (0)