Caffe：我可以在训练时避免加载新数据吗？

Question

我想在一小批数据上装配我的模型，看看我的模型是否正确。我的数据集是lmdb格式。执行solver.step（）时，数据层会自动更新。如何避免求解器在Caffe中加载新数据？

Answer 1

我在Pycaffe界面使用它：

if overfit：lmdb_cursor.first（）

我有一个标志（overfit），当set设置调用此方法时，会将光标重置为每个批次的数据库的开头。希望这会有所帮助。