nni icon indicating copy to clipboard operation
nni copied to clipboard

Error in pruning

Open N-Kingsley opened this issue 3 years ago • 2 comments

I try to use l1norm pruner to prune the bert classification model, but get the error as below: image

N-Kingsley avatar Jun 09 '22 03:06 N-Kingsley

Thanks for your finding, nni doesn't support replace torch.nn.Embedding right now, we will try to support it as soon. But I think this place seems to have additional problems, could you give us a simple example to reproduce your problem?

J-shang avatar Jun 10 '22 02:06 J-shang

Thanks for your finding, nni doesn't support replace torch.nn.Embedding right now, we will try to support it as soon. But I think this place seems to have additional problems, could you give us a simple example to reproduce your problem?

@N-Kingsley - would you mind provide more info for reproducing the problem? thanks.

scarlett2018 avatar Jun 15 '22 02:06 scarlett2018