TypeError: cannot assign 'torch.cuda.FloatTensor' as parameter 'weight' (torch.nn.Parameter or None expected) about torch-model-compression HOT 6 CLOSED

Show me the source code, I will check where the error is.

from torch-model-compression.

Hello, you may try the patch in https://github.com/THU-MIG/torch-model-compression/pull/6/files#diff-cd7d0b02724faaf1716aee410c2cc83b91a053f1f6e8dc9475761156a69a6286R50-R52

from torch-model-compression.

I've merged #6, so you may run git pull on your local repo and reinstall it.

from torch-model-compression.

I've merged #6, so you may run git pull on your local repo and reinstall it.

Thanks for your in-time reply. I tried to pull the latest code and retrain, but got such error message:

Traceback (most recent call last):
File "tasks/prune_helmet.py", line 158, in
solver.run()
File "/home/bengui/miniconda3/envs/torch1.7/lib/python3.6/site-packages/torchpruner-0.0.1-py3.6.egg/torchslim/slim_solver.py", line 327, in run
File "/home/bengui/miniconda3/envs/torch1.7/lib/python3.6/site-packages/torch/optim/lr_scheduler.py", line 67, in wrapper
return wrapped(*args, **kwargs)
File "/home/bengui/miniconda3/envs/torch1.7/lib/python3.6/site-packages/torch/autograd/grad_mode.py", line 26, in decorate_context
return func(*args, **kwargs)
File "/home/bengui/miniconda3/envs/torch1.7/lib/python3.6/site-packages/torch/optim/sgd.py", line 106, in step
buf.mul_(momentum).add_(d_p, alpha=1 - dampening)
RuntimeError: The size of tensor a (96) must match the size of tensor b (88) at non-singleton dimension 0

from torch-model-compression.

pull the latest code, and try it again. The error has been fixed.

from torch-model-compression.

pull the latest code, and try it again. The error has been fixed.

I tried it again and the error had been fixed.

from torch-model-compression.