文件名称：channel-prune
文件大小：22KB
文件格式：ZIP
更新时间：2024-06-05 01:10:33
pytorch model-pruning Python
这是的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下定义一个转发函数和
【文件预览】：
channel-prune-master
----models()
--------__init__.pyc(142B)
--------inception_res_v1.py(9KB)
--------__init__.py(0B)
--------inception_res_v2.py(11KB)
----model_prune()
--------filter_pruner.py(10KB)
--------prune_utils.py(3KB)
--------__init__.py(0B)
----prune_Resnet50_example.py(11KB)
----prune_InceptionV3_example.py(32KB)
----prune_IncepResV1_example.py(32KB)
----.idea()
--------vcs.xml(180B)
----.gitignore(1KB)
----README.md(1KB)
----utils()
--------logs.py(442B)
--------train.py(4KB)
--------__init__.py(0B)