AttributeError: 'MagnitudePruner' object has no attribute 'pruning_history'

[chenyuuuma]

当我尝试用pruning_history来存储剪枝后模型时报错！我的代码是按照test_load.py写的，请问是版本问题吗？

[daixiangzi]

+1

[VainF]

您好，请尝试一下https://github.com/VainF/Torch-Pruning/blob/master/tests/test_serialization.py，和pruning_history类似但是更简洁。pruning_history的问题我待会捣鼓一下。