shicai / DenseNet-Caffe

DenseNet Caffe Models, converted from https://github.com/liuzhuang13/DenseNet
357 stars 273 forks source link

revise deploy.prototxt #9

Closed xizi closed 7 years ago

xizi commented 7 years ago

Hi, shicai, can i directly revise deploy.prototxt to get trainval.prototxt and what is the need to pay attention to detail?

shicai commented 7 years ago

yes.

yuanyuanli85 commented 7 years ago

I think you can refer to code to generate deploy and train_val prototxt. https://github.com/liuzhuang13/DenseNetCaffe/ or https://github.com/yuanyuanli85/DenseNet-Caffe

dlyldxwl commented 7 years ago

请问一下,为什么bn层需要加3次呢,不是只有两个参数吗,而且这两个参数也是不用学习的呀

shicai commented 7 years ago

caffe默认的batchnorm层有3个参数,正是因为这些参数是需要计算而不是需要训练的,所以才需要将lr_mult和decay_mult设置为0。

dlyldxwl commented 7 years ago

谢谢回答!我看了一下源码,第三个参数大概是滑动系数和的意思,前两个分别是mean的和、var的和。但是我看了一下你的densenet model为啥第三个系数值是1呢,你做了reshape吗?

dlyldxwl commented 7 years ago

错了,我可视化的是你的nobilenet model的conv1/bn里的第三个参数值是1。你训练的model很赞,给力!

shicai commented 7 years ago

thx~