Какие веса итераций сохраняются для развертывания, тестирования?

Question

Я тренирую unet нейронную сеть. Во время обучения каждая итерация имеет «значение потерь». Это значение обычно сходится, но иногда перепрыгивает. Какие веса наконец сохраняются в файле .caffemodel?

Что произойдет, если я сохраню его на итерации 20000, и именно в этот момент потери немного подскочили, и это не самая низкая потеря, которую он видел? Веса и уклоны сохраняются после последней итерации или что-то более умное, например, самая низкая из последних 5% итераций?

Спасибо

Answer 1

Solver.prototxt имеет один параметр с именем «снимок»

net: "path/to/train.prototxt"
.
.
max_iter: 20000
snapshot: 1000
snapshot_prefix: "path/to/caffemodel/"
solver_mode: GPU

Например, если вы исправите снимок: 1000, то каждые 1000 итераций будет сохраняться один файл .caffemodel с весами, соответствующими этому итерация, независимо от того, была ли потеря меньше на предыдущей итерации.