added safeguards for callbacks in loading saving

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