Python 3.6, pytorch 0.4.1, getting RuntimeError: Dimension out of range (expected to be in range of [-2, 1], but got 3) about pytorch-pruning HOT 4 OPEN

pytorch version caused the problem, you can change to this: values =
torch.sum((activation * grad), dim=0, keepdim = True).
sum(dim=2, keepdim = True).sum(dim=3,keepdim = True)[0, :, 0, 0].data

from pytorch-pruning.

@wuzhiyang2016 I used keepdim=True, but I still get errors at the same point.

RuntimeError: The size of tensor a (14) must match the size of tensor b (28) at non-singleton dimension 3
Any idea?

Thanks!

from pytorch-pruning.

@wuzhiyang2016 I used keepdim=True, but I still get errors at the same point.

RuntimeError: The size of tensor a (14) must match the size of tensor b (28) at non-singleton dimension 3
Any idea?

Thanks!

what's your pytorch version? and your error is not "Dimension out of range (expected to be in range of [-2, 1], but got 3)"

from pytorch-pruning.

@bidakos Have you solved this problem? I need your help and I met same issue, pls help me

from pytorch-pruning.