From b684fdf502212f91a033f14da03a484d588bc2ac Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 10:55:17 -0400
Subject: [PATCH] removed dummy d

---
 tests/debug.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tests/debug.py b/tests/debug.py
index 1375930d91..49483e0bbb 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -62,6 +62,14 @@ def main():
     # correct result and ok accuracy
     assert result == 1, 'amp + ddp model failed to complete'
 
+    # test prediction
+    data = model.test_dataloader
+    for batch in data:
+        break
+    out = model(data[0])
+    print(out)
+
+
     clear_tt_dir()