From 3fc8166f51badbe36dde17817df1474020223b4b Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 11:35:55 -0400
Subject: [PATCH] added safeguards for callbacks in loading saving

---
 tests/debug.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/debug.py b/tests/debug.py
index 7f97d86dfe..c0d845c50e 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -96,7 +96,7 @@ def main():
     x, y = batch
     x = x.view(x.size(0), -1)
 
-    y_hat = model(x)
+    y_hat = trained_model(x)
 
     # acc
     labels_hat = torch.argmax(y_hat, dim=1)