From 00d5285ce41b6965b9fe51479072e4ec78bf682e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 16:50:54 -0500 Subject: [PATCH] finished lightning module --- pytorch_lightning/callbacks/pt_callbacks.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/pytorch_lightning/callbacks/pt_callbacks.py b/pytorch_lightning/callbacks/pt_callbacks.py index 117c4706..72d41abf 100644 --- a/pytorch_lightning/callbacks/pt_callbacks.py +++ b/pytorch_lightning/callbacks/pt_callbacks.py @@ -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',