Updated to work with pytorch 0.2.

Weihnachtswurst · Weihnachtswurst · commit 6d989b9de981 · 2017-09-09T07:26:00.000-07:00
diff --git a/demo.py b/demo.py
@@ -31,7 +31,6 @@
 preds = model(image)
 
 _, preds = preds.max(2)
-preds = preds.squeeze(2)
 preds = preds.transpose(1, 0).contiguous().view(-1)
 
 preds_size = Variable(torch.IntTensor([preds.size(0)]))
