MCPcopy Create free account
hub / github.com/alibaba/MNN / setLearningRate

Method setLearningRate

tools/train/source/optimizer/SGD.cpp:22–24  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

20}
21
22void SGD::setLearningRate(float rate) {
23 mLearningRate = rate;
24}
25
26void SGD::setMomentum(float momentum) {
27 mMomentum = momentum;

Callers 14

_quantizeModelEMAMethod · 0.80
trainMethod · 0.80
runMethod · 0.80
_trainFunction · 0.80
_trainFunction · 0.80
trainMethod · 0.80
runMethod · 0.80
runMethod · 0.80
runMethod · 0.80
runMethod · 0.80
runMethod · 0.80
createSGDMethod · 0.80

Calls

no outgoing calls

Tested by 5

runMethod · 0.64
runMethod · 0.64
runMethod · 0.64
runMethod · 0.64
runMethod · 0.64