wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

finished lightning module

Browse Source
This commit is contained in:
William Falcon
2020-01-16 16:50:54 -05:00
parent 070d008bd0
commit 00d5285ce4
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/callbacks/pt_callbacks.py
+7 -5
View File
@@ -86,12 +86,14 @@ class EarlyStopping(Callback):
monitored has stopped increasing; in `auto`
mode, the direction is automatically inferred
from the name of the monitored quantity.
Example:
>>> from pytorch_lightning import Trainer
>>> from pytorch_lightning.callbacks import EarlyStopping
>>>
>>> early_stopping = EarlyStopping('val_loss')
>>> Trainer(early_stop_callback=early_stopping)
from pytorch_lightning import Trainer
from pytorch_lightning.callbacks import EarlyStopping
early_stopping = EarlyStopping('val_loss')
Trainer(early_stop_callback=early_stopping)
"""
def __init__(self, monitor='val_loss',