cmd23333 / Faster-rcnn-tensorflow2.x

An implement of origin faster-rcnn based on tensorflow 2.x and visualize it.使用Tensorflow2.x实现了faster-rcnn并可视化算法运行流程
46 stars 12 forks source link

请问支持多GPU并行训练吗? #4

Open HuizhouLi opened 4 years ago

cmd23333 commented 4 years ago

暂时不支持 最近我也比较忙,没空改进(前两个issue都还没解决呢 = =。) 这个项目其实是为了让大家更好的理解这个算法而建立的,大型的开源框架都整合了太多了功能,我只保留了最核心的部分。 但真正应用这个算法,类似多GPU啊/batch size大于1啊,等等,还是推荐用大公司开源的框架[maskrcnn-benchmark就不错](其实我这边的代码效率比较低的)

HuizhouLi commented 4 years ago

暂时不支持 最近我也比较忙,没空改进(前两个issue都还没解决呢 = =。) 这个项目其实是为了让大家更好的理解这个算法而建立的,大型的开源框架都整合了太多了功能,我只保留了最核心的部分。 但真正应用这个算法,类似多GPU啊/batch size大于1啊,等等,还是推荐用大公司开源的框架[maskrcnn-benchmark就不错](其实我这边的代码效率比较低的)

好的,谢谢~