基于pytorch的模型稀疏訓練與模型剪枝示例

稀疏訓練+模型剪枝代碼下載地址：下載地址

CIFAR10-VGG16BN	Baseline	Trained with Sparsity (1e-4)	Pruned (0.7 Pruned)	Fine-tuned (40epochs)
Top1 Accuracy (%)	93.62	93.77	10.00	93.56
Parameters	20.04M	20.04M	2.42M	2.42M

Pruned Ratio	0	0.1	0.2	0.3	0.4	0.5	0.6

總結

以上是生活随笔為你收集整理的基于pytorch的模型稀疏训练与模型剪枝示例的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。