Caffe: Могу ли я избежать загрузки новых данных во время обучения?

Question

Я хочу переделать свою модель на мини-пакет данных, чтобы убедиться, что моя модель верна. Мой набор данных находится в формате lmdb. Уровень данных автоматически обновляется при выполнении solver.step(). Как я могу избежать решателя от загрузки новых данных в Caffe?

Answer 1

Я использую это с интерфейсом Pycaffe:

, если overfit: lmdb_cursor.first()

У меня есть флаг (overfit), что, когда множество вызывает этот метод, который сбрасывает курсор обратно к началу база данных для каждой партии. Надеюсь это поможет.