caffe训练数据中出现[data_layer.cpp:73] Restarting data prefetching from start. 的问题

        一开始在网上查了许多的,有人回答说是之前的一些prototxt文件参数设置有问题,还有人说是batch_size相对于整个数据集太大的问题。

        我一开始也以为这是个warning提示,会对训练结果产生影响,但是稳步上升的accuracy,又觉得是没问题的。

        于是我现在觉得,这个提示,仅仅是个提示,对训练结果是没有任何影响的。

        我的看法是,这个提示仅仅说明遍历了一次训练集的训练输入。因此我之前的训练集有4400个,batch_size设为44,test_iter设置为100,而且设置了display为100,因此每训练100次,就遍历完整个训练集,便会提示一次,restarting,同时,也是我选择display一次的时机。

        为了验证这个想法,我将display设为400,即每400次才打印一次训练的loss等结果,想法正确的话,应该会每显示4次restarting后打印一次。

        我觉得这就验证成功了,当然,有时间还是看源代码比较准确,可惜最近忙,没时间,算了,有时间再看看。

猜你喜欢

转载自blog.csdn.net/weixin_39220714/article/details/84880816
今日推荐