Shape mismatch after compression/speedup

Open lminer opened this issue 2 years ago • 0 comments

After running AGPPRUNER followed by model speedup, I'm getting a shape mismatch in a Linear layer. The weight and bias in the linear layer don't appear to be of matching size anymore:

Environment:

NNI version: 3.0rc1
Training service (local|remote|pai|aml|etc):
Python version: 3.10
PyTorch version: 2.0
Cpu or cuda version: 11.7

Reproduce the problem

Code|Example:
How to reproduce:

Aug 25 '23 22:08 lminer