I've converted a bunch of .py files to Python 3 and am getting now:
(py35) ~/yolo2-pytorch$ python demo.py
load model succ...
Traceback (most recent call last):
File "demo.py", line 49, in <module>
bbox_pred, iou_pred, prob_pred = net(im_data)
File "/home/wilrich/anaconda3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 202, in __call__
result = self.forward(*input, **kwargs)
File "/home/wilrich/yolo2-pytorch/darknet.py", line 177, in forward
conv1s_reorg = self.reorg(conv1s)
File "/home/wilrich/anaconda3/lib/python3.5/site-packages/torch/nn/modules/module.py", line 202, in __call__
result = self.forward(*input, **kwargs)
File "/home/wilrich/yolo2-pytorch/layers/reorg/reorg_layer.py", line 49, in forward
x = ReorgFunction(self.stride)(x)
File "/home/wilrich/yolo2-pytorch/layers/reorg/reorg_layer.py", line 15, in forward
out = torch.FloatTensor(bsize, out_c, out_h, out_w)
TypeError: torch.FloatTensor constructor received an invalid combination of arguments - got (int, int, float, float), but expected one of:
* no arguments
* (int ...)
didn't match because some of the arguments have invalid types: (int, int, float, float)
* (torch.FloatTensor viewed_tensor)
* (torch.Size size)
* (torch.FloatStorage data)
* (Sequence data)
Hi,
I've converted a bunch of .py files to Python 3 and am getting now:
Thanks for having a look, wr