From 9ff6108af1b8c4a2aec8d65ce785a85ab881fe77 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sun, 31 Mar 2019 16:34:13 -0400
Subject: [PATCH] added example and verified

---
 README.md             | 15 ++++++---------
 demo/basic_trainer.py | 12 +++++-------
 2 files changed, 11 insertions(+), 16 deletions(-)

diff --git a/README.md b/README.md
index 245004c2..6a1afe59 100644
--- a/README.md
+++ b/README.md
@@ -51,8 +51,6 @@ To use lightning do 2 things:
 #### Define the trainer   
 
 ```python
-# trainer.py
-
 import os
 import sys
 
@@ -83,16 +81,14 @@ def main(hparams):
     exp.save()
 
     # build model
-    print('loading model...')
     model = ExampleModel(hparams)
-    print('model built')
 
     # callbacks
     early_stop = EarlyStopping(
-        monitor=hparams.early_stop_metric,
-        patience=hparams.early_stop_patience,
+        monitor='val_acc',
+        patience=3,
+        mode='min',
         verbose=True,
-        mode=hparams.early_stop_mode
     )
 
     model_save_path = '{}/{}/{}'.format(hparams.model_save_path, exp.name, exp.version)
@@ -101,8 +97,8 @@ def main(hparams):
         save_function=None,
         save_best_only=True,
         verbose=True,
-        monitor=hparams.model_save_monitor_value,
-        mode=hparams.model_save_monitor_mode
+        monitor='val_acc',
+        mode='min'
     )
 
     # configure trainer
@@ -129,6 +125,7 @@ if __name__ == '__main__':
 
     # train model
     main(hyperparams)
+
 ```
 
 #### Define the model   
diff --git a/demo/basic_trainer.py b/demo/basic_trainer.py
index 0ce8a28b..95a47dd6 100644
--- a/demo/basic_trainer.py
+++ b/demo/basic_trainer.py
@@ -28,16 +28,14 @@ def main(hparams):
     exp.save()
 
     # build model
-    print('loading model...')
     model = ExampleModel(hparams)
-    print('model built')
 
     # callbacks
     early_stop = EarlyStopping(
-        monitor=hparams.early_stop_metric,
-        patience=hparams.early_stop_patience,
+        monitor='val_acc',
+        patience=3,
+        mode='min',
         verbose=True,
-        mode=hparams.early_stop_mode
     )
 
     model_save_path = '{}/{}/{}'.format(hparams.model_save_path, exp.name, exp.version)
@@ -46,8 +44,8 @@ def main(hparams):
         save_function=None,
         save_best_only=True,
         verbose=True,
-        monitor=hparams.model_save_monitor_value,
-        mode=hparams.model_save_monitor_mode
+        monitor='val_acc',
+        mode='min'
     )
 
     # configure trainer