dirichlet_pruning 源码
###运行Lenet压缩; 1.首先训练模型 python lenet_network_pruning.py --early_stopping 30 然后提供训练模式的名称 2.修剪: 示例(从步骤1中选择您自己的模型): python lenet_network_pruning.py --resume=True --prune_bool=True --path mnist_trainval0.8_epo8_acc98.70 3.修剪和再培训: python lenet_network_pruning.py --resume=True --prune_bool=True --path mnist_trainval0.8_epo8_acc98.70 --retrain=True 或者 python lenet_network_pruning.py --resume=True --pru
文件列表
dirichlet_pruning-master.zip
(预估有个39文件)
dirichlet_pruning-master
Resnet56
trainer_prune.py
17KB
models
resnet2.py
5KB
vgg.py
0B
lenet5.py
1KB
__init__.py
0B
resnet.py
5KB
VGG
暂无评论