1. 程式人生 > >caffe中的iter_size是什麼?和batch_size什麼關係 ?

caffe中的iter_size是什麼?和batch_size什麼關係 ?

iter_size :

這個引數乘上你的train prototxt中的batch_size是你實際使用的batch size。 相當於讀取batchsize*itersize個影象才做一下gradient decent。 這個引數可以規避由於gpu不足而導致的batchsize的限制 因為你可以用多個iteration做到很大的batch 即使單次batch有限

參考資料:

Caffe中的batch_size和iter_size
caffe SolverParameter中的iter_size引數什麼作用?