Yolo模型训练时不收敛调试

发布时间:2017-11-01  栏目:人工智能, 机器视觉  评论:0 Comments

在训练二值化之后的Tiny Yolo模型时,模型不收敛,观察IOU值迅速向两边扩散,解决方法:

(1)调小Learning rate以及momentum参数,观察实时训练结果进行调整

(2)加大训练的迭代次数,因为二值化之后的模型需要更长的时间才能收敛

留下评论

You must be logged in to post a comment.

相册集

pix pix pix pix pix pix

关于自己

杨文龙,微软Principal Engineering Manager, 曾在各家公司担任影像技术资深总监、数据科学团队资深经理、ADAS算法总监、资深深度学习工程师等职位,热爱创新发明,专注于人工智能、深度学习、图像处理、机器学习、算法、自然语言处理及软件等领域,目前发明有国际专利19篇,中国专利28篇。

联系我

个人技术笔记

welonshen@gmail.com

2015 in Shanghai