removed dummy d

2019-07-24 10:57:46 -04:00 · 2019-07-24 10:57:46 -04:00 · 88f064d276
parent eb4b3a5752
commit 88f064d276
1 changed files with 4 additions and 1 deletions
--- a/tests/debug.py
+++ b/tests/debug.py
@ -66,7 +66,10 @@ def main():
    data = model.test_dataloader
    for batch in data:
        break
-    out = model(batch[0])
+
+    x, y = batch
+    x = x.view(x.size(0), -1)
+    out = model(x)
    print(out)