PytorchSSD icon indicating copy to clipboard operation
PytorchSSD copied to clipboard
verified publisher icon lzx1413

执行 demo/live.py出错

Open moondaiy opened this issue 7 years ago • 1 comments

你好,我在执行demo/live.py 文件时, 发生了错误. Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 'weight' 我目前pytorch版本是0.3.0

moondaiy avatar Aug 13 '18 02:08 moondaiy

需要做一个模型从gpu到cpu的转换，如果是在cpu上跑的话

lzx1413 avatar Aug 29 '18 02:08 lzx1413