如何加快caffe中的图像加载？

Question

在使用caffe训练CNN时，在打印出的每次迭代之间，我得到3到5次重复的以下警告：

blocking_queue.cpp:50] Waiting for data

我正在使用默认的caffe设置，因为我使用提供的脚本创建了一个lmdb数据库。在加载到lmdb之前，图像是200x200px tif，大约30-50kB。

我该如何诊断减速的位置？即硬件链的哪个部分需要升级，或者我可以用软件来加速处理。我正在使用带有CUDA版本7.5.18的GeForce GTX 980 GPU。

Answer 1

这可能是您的I / O链的问题。您是否有应用程序来监控设备的I / O功能？在“等待数据”的同时是否有缓慢的响应？

一种可能的解决方法是增加批量大小：一次读入更多数据将使模型更长时间处于繁忙状态，并为I / O例程提供更多时间来获取下一个数据块。