finetune踩坑

最近在学习深度学习网络,在MobileNetv3+YUNYANG1994大神的yolov3
使用自己的数据集从零训练没有任何问题。
但是使用Google官方给出基于imagenet训练的预训练模型finetune的时候,trainloss总是维持在18.5降不下去。
几次尝试和摸索,总结几点(问题不分先后):
1.finetune分为两种,(一)局部微调,(二)全局微调,局部微调的时候,需要注意,在用优化器优化的时候,哪些参数是不变,哪些参数是变得。
2.各个部分的参数名字要与函数中的一一对应。
3.预训练模型中有的参数的值直接继承过来就好,预训练模型中没有的参数,要根据yolo模型中各层原本的初始化参数进行初始化,比如yolo中的weight是‘glorot_uniform’,bias是‘zeros’,beta和moving_mean是‘zeros’,gamma和moving_variance是‘Ones’,将带有ExponentialMovingAverage的都初始化为相反的 等等。

问题3的发现是将trainloss=18.5的模型解析,ckpt文件中,yolo模型部分参数都是0或无限接近0的数(因为在初始化的时候,将所有的都统一初始化为0或random_uniform了)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值