问题描述:
我在网上找到了一种LSTM的剪枝方法——Topk剪枝方法。我现在希望评测它的性能指标。我希望使用不同的模型和不同的数据集去评测它。但是这个剪枝方法原本是针对LSTM设计的,我怀疑它是否能用到像vgg.resnet这种CNN上去;而且LSTM原本是语言模型,我也不确定它是否能使用ImageNet和cifar这种数据集去训练。这个大家有什么思路或建议吗?
PS:我在GitHub上看到有个叫torch_pruning的包,里面提出了depgraph的技术,我理解它是一种媒介,就是任何剪枝方法可以通过这个媒介去修剪任意的模型。我也不知道理解得对不对。希望各位批评指正。谢谢!