leeyoshinari / YOLO_v2

The implementation of YOLO v2 with TensorFlow
GNU General Public License v3.0
80 stars 38 forks source link

关于预训练模型的问题 #38

Closed QuietDeer closed 4 years ago

QuietDeer commented 5 years ago

谢谢作者的分享! 不过我有个问题,我不导入那个模型直接训练的话就会梯度爆炸,导入的话就比较正常,请问您的预训练的模型是怎么得到的?我想学一下

leeyoshinari commented 5 years ago

你好,可以改变learning_rate、batch_size、权值初始化方法、梯度下降算法等来动态调整训练过程,也可以时模型收敛,但效果不是很好。其实我这个预训练模型也是迁移学习得到的。