shiuang

Here is a simple case I met which had the same error. I hope that can help you guys to debug. ``` import torch import torch.nn as nn from nni.compression.pytorch.pruning...