search

ArtyZe / yolov3_lite

yolov3 model compress and acceleration (quantization, sparse)， c++ version
37 stars 9 forks source link

L1 scales regularization #2

Open andylai0212 opened 5 years ago

andylai0212 commented 5 years ago

你好，感谢您的分享。 請問可以說明一下關於L1 scales regularization 方法具體需要修改源碼那些地方嗎?

ArtyZe commented 5 years ago

@andylai0212 你好，就修改convolutional_layer_kernels.cu中，你可以看到我有一个l1正则化的宏定义，SCALE_L1，你可以搜这个关键字，所有有这个的都是

ujsyehao commented 5 years ago

@ArtyZe 请教下你这个L1正则化是针对conv weights还是针对BN的scale值做的？我正在实验一个模型剪枝方法，参考的是这篇文章

Learning efficient convolution networks through network slimming(ICCV 2017)