基于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量化感知训练,稀疏训练,模型剪枝学习教程,自动驾驶,人工智能,机器学习)