From 88f064d276868b70961bc7f80265b5ef33666f70 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 10:57:46 -0400 Subject: [PATCH] removed dummy d --- tests/debug.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/debug.py b/tests/debug.py index 14d1e56b56..a7dd56aa14 100644 --- 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)