Tensorflow目标检测API在MSCOCO数据集上的训练时间

4

使用Tensorflow目标检测API在单个GPU卡(如K40或1080)上训练MSCOCO Faster RCNN模型通常需要多长时间?如果您还能提供其他模型(SSD和R-FCN)的训练时间,那就太好了。

1个回答

4

我从未使用单个GPU在COCO上训练模型,通常我们会使用大约10个k40 GPU使用异步SGD进行训练,在COCO上收敛需要3-4天时间。SSD和R-FCN需要相同的时间。


你认为单个Tesla P100使用SSD/Faster-RCNN需要多长时间?我使用的数据集是自定义的,有34个类别和70k个物体图像。对于时间或迭代次数有任何大致的想法吗?400k?1m?10m? - john doe

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接