35 lines
796 B
ReStructuredText
35 lines
796 B
ReStructuredText
|
Early stopping
|
||
|
|
==================
|
||
|
|
|
||
|
|
|
||
|
|
Enable Early Stopping
|
||
|
|
----------------------
|
||
|
|
There are two ways to enable early stopping.
|
||
|
|
|
||
|
|
.. note:: See: :ref:`trainer`
|
||
|
|
|
||
|
|
.. code-block:: python
|
||
|
|
|
||
|
|
# A) Looks for val_loss in validation_step return dict
|
||
|
|
trainer = Trainer(early_stop_callback=True)
|
||
|
|
|
||
|
|
# B) Or configure your own callback
|
||
|
|
early_stop_callback = EarlyStopping(
|
||
|
|
monitor='val_loss',
|
||
|
|
min_delta=0.00,
|
||
|
|
patience=3,
|
||
|
|
verbose=False,
|
||
|
|
mode='min'
|
||
|
|
)
|
||
|
|
trainer = Trainer(early_stop_callback=early_stop_callback)
|
||
|
|
|
||
|
|
Force disable early stop
|
||
|
|
-------------------------------------
|
||
|
|
To disable early stopping pass None to the early_stop_callback
|
||
|
|
|
||
|
|
.. note:: See: :ref:`trainer`
|
||
|
|
|
||
|
|
.. code-block:: python
|
||
|
|
|
||
|
|
# DEFAULT
|
||
|
|
trainer = Trainer(early_stop_callback=None)
|