added safeguards for callbacks in loading saving

2019-07-24 11:35:55 -04:00 · 2019-07-24 11:35:55 -04:00 · 3fc8166f51
parent 2e30dd94bc
commit 3fc8166f51
1 changed files with 1 additions and 1 deletions
--- 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)