Merge pull request meijieru#61 from Weihnachtswurst/master

meijieru · web-flow · commit ab61ff13c204 · 2017-09-10T13:59:09.000+08:00
Updated to work with pytorch 0.2.
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)]))
