File tree 1 file changed +5
-1
lines changed
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 275
275
net .params(i ).der = min(max(net .params(i ).der, - opts .gradRange / mult ), opts .gradRange / mult );
276
276
thisDecay = opts .weightDecay * net .params(i ).weightDecay ;
277
277
278
+ momentum_prev = state.momentum{i };
278
279
state.momentum{i } = opts .momentum * state.momentum{i } ...
279
280
- lr * net .params(i ).learningRate * ...
280
281
thisDecay * net .params(i ).value ...
281
282
- lr * net .params(i ).learningRate * (1 / batchSize ) * net .params(i ).der ;
282
283
283
- net .params(i ).value = net .params(i ).value + state.momentum{i };
284
+ % Nesterov
285
+ net .params(i ).value = net .params(i ).value ...
286
+ - opts .momentum * momentum_prev ...
287
+ + (1 + opts .momentum ) * state.momentum{i };
284
288
end
285
289
286
290
% -------------------------------------------------------------------------
You can’t perform that action at this time.
0 commit comments