From 58d52c25a1618a768ee48b4f75b2f70e7fa89e2f Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 19 Oct 2019 00:51:48 +0200 Subject: [PATCH] Fixes #347 (#393) --- docs/Trainer/Training Loop.md | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/docs/Trainer/Training Loop.md b/docs/Trainer/Training Loop.md index b8ccfc85..abb4d920 100644 --- a/docs/Trainer/Training Loop.md +++ b/docs/Trainer/Training Loop.md @@ -35,15 +35,22 @@ early_stop_callback = EarlyStopping( 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) + +# pass in None to disable it +trainer = Trainer(early_stop_callback=None) ``` --- #### 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} # DEFAULT -trainer = Trainer(enable_early_stop=True) +trainer = Trainer(early_stop_callback=None) ``` ---