这是的PyTorch实现。 在可以实现3倍的模型尺寸缩减,并且精度损失很小。 该项目是从修改而来的,主要区别是: 修剪一次完成,而不是顺序执行。 所引用的原始项目中的FilterPruner和PruningFineTuner被合并到FilterPruner类中,以使其更加简洁。 修剪InceptionV3 , Inception_Resnet_V1 , Resnet50作为示例,您可以为自定义模型定义新的FilterPruner。 笔记: 您可以使用PyTorch的预训练的Resnet50或InceptionV3作为基本模型,并在前面提到的cat-vs-dog数据集中修剪它们。 (请参阅prune_InceptionV3_example.py和prune_Resnet50_example.py) 要修剪新模型,您需要根据模型的体系结构在FilterPruner下定义一个转发函数和
2021-11-08 13:19:31 22KB pytorch model-pruning Python
1