基于pytorch的模型稀疏训练与模型剪枝示例
生活随笔
收集整理的這篇文章主要介紹了
基于pytorch的模型稀疏训练与模型剪枝示例
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
基于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的模型稀疏训练与模型剪枝示例的全部內容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： 基于YOLO的密集人脸检测（课程设计）
- 下一篇： 【camera】基于YOLO的车辆多维特