lightning/docs/source/early_stopping.rst

Early stopping
==============

Default behavior
----------------
By default early stopping will be enabled if `'val_loss'`
is found in `validation_epoch_end()` return dict. Otherwise
training will proceed with early stopping disabled.

Enable Early Stopping
---------------------
There are two ways to enable early stopping.

.. seealso:: :ref:`trainer`

.. code-block:: python

    # A) Set early_stop_callback to True. Will look for 'val_loss'
    # in validation_epoch_end() return dict. If it is not found an error is raised.
    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)

Disable Early Stopping
----------------------
To disable early stopping pass ``False`` to the `early_stop_callback`.
Note that ``None`` will not disable early stopping but will lead to the
default behaviour.

.. seealso:: :ref:`trainer`