We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a300fe2 commit 8ca5695Copy full SHA for 8ca5695
src/lightning/pytorch/callbacks/early_stopping.py
@@ -90,7 +90,8 @@ class EarlyStopping(Callback):
90
Example::
91
92
>>> from lightning.pytorch import Trainer
93
- >>> from lightning.pytorch.callbacks import EarlyStopping, EarlyStoppingReason
+ >>> from lightning.pytorch.callbacks import EarlyStopping
94
+ >>> from lightning.pytorch.callbacks.early_stopping import EarlyStoppingReason
95
>>> early_stopping = EarlyStopping('val_loss')
96
>>> trainer = Trainer(callbacks=[early_stopping])
97
>>> # After training...
0 commit comments