minibatch Iterations solver.step

9 views
Skip to first unread message

Roger Trullo

unread,
Jul 1, 2016, 8:14:25 AM7/1/16
to Caffe Users
Hi,
I am wondering,
when I do solver.step(1),
I understand this is a minibatch update in SGD.
so if I want a loop in epochs, that is, one iteration the whole data set, should I do something like

iter_epochs=nbatches

for iter=1:nbatches*iterations
solver.step(1)
end

or this:

for iter=1:iterations
solver.step(nbatches)
end

Thanks in advance.


Reply all
Reply to author
Forward
0 new messages