Attribute Error: module ‘torch_pruning‘ has no attribute ‘prune_conv‘

Attribute Error: module ‘torch_pruning’ has no attribute 'prune_conv’问题解决

这个问题很好解决,就是库版本对不上的原因。目前如果安装最新版本torch_pruning=0.2.8就会出现这个问题,可能是prune_conv被弃用了或者被其他的模块代替了。
直接把版本改成0.2.4即可解决问题

pip uninstall torch_pruning
pip install torch_pruning==0.2.4

你可能感兴趣的:(剪枝,python,深度学习)