From 57a99e2aa5ad6d4b192dae24a3c352d970b6df37 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 10:30:41 -0400
Subject: [PATCH] updated test docs

---
 pytorch_lightning/models/trainer.py | 2 ++
 tests/debug.py                      | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index ae5eb55c..62df9ef4 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -254,6 +254,8 @@ class Trainer(TrainerIO):
 
     @property
     def __tng_tqdm_dic(self):
+        import pdb
+        pdb.set_trace()
         tqdm_dic = {
             'tng_loss': '{0:.3f}'.format(self.avg_loss),
             'v_nb': '{}'.format(self.experiment.version),
diff --git a/tests/debug.py b/tests/debug.py
index facb7b62..42b5ef75 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -60,6 +60,8 @@ def main():
     # correct result and ok accuracy
     assert result == 1, 'amp + ddp model failed to complete'
 
+    trainer.tng_tqdm_dic
+
     clear_tt_dir()
 
 if __name__ == '__main__':