文件名称：caffe-admm:使用乘法器交替方向方法（ADMM）的DNN连接修剪
文件大小：8.44MB
文件格式：ZIP
更新时间：2024-05-17 08:21:46
C++
安装 与Caffe的原始版本相同。 有关最新的安装说明，请参见 。 如何训练 修剪神经网络需要两个步骤。 假设我们有一个预先训练的模型“ bvlc_alexnet.caffemodel”。 步骤1：ADMM（抑制目标体重） ./build/tools/caffe train --solver=models/bvlc_alexnet/solver_admm.prototxt --weights=/path/to/bvlc_alexnet.caffemodel 第2步：重新训练（将目标权重设置为0并重新训练） ./build/tools/caffe train --solver=models/bvlc_alexnet/solver_retrain.prototxt --weights=models/bvlc_alexnet/caffe_alexnet_train_admm_iter_2