Giters

liuzhuang13 / slimming

Learning Efficient Convolutional Networks through Network Slimming, In ICCV 2017.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

在剪枝程序vggprune.py中遇见程序报错

xuemo188 opened this issue · comments

commented

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'