剪枝程序vggprune.py遇见的问题

[xuemo188]

Traceback (most recent call last):
File "vggprune.py", line 73, in
mask = weight_copy.gt(thre).float().cuda()
RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #2 'other'
我用的环境是python3.6 torch0.4.1

[Thea1]

Have you solved the problem? I encountered the same problem as you.

[YuQi9797]

Have you solved the problem? I encountered the same problem as you.

你们需要将其前面的thre变成cuda() 即 thre.cuda()，后再进行该mask的计算。