Keras多GPU训练问题解决汇总-多GPU-速度慢-单GPU无法加载权重文件-pretrain问题解决
我使用的是qqwweee/keras-yolo3的代码,代码地址目前用于做某个设备的工业化检测,实现召回率98%以上,误检在2%左右,已满足应用需求。解决了4个问题:1.多GPU训练问题;2.多GPU训练速度慢的问题;3.单GPU无法加载多GPU跑出的加载权重问题;4.Pretrain训练收敛速度慢的问题。1.多GPU训练问题参考的是这篇文章①首先在train....
原创
2019-10-30 15:02:32 ·
4105 阅读 ·
7 评论