How to change learning rate and momentum in 'trainlm' neural network?

조회 수: 6 (최근 30일)
Avanthi Saumyamala
Avanthi Saumyamala 2018년 11월 2일
댓글: Katy 2023년 7월 19일
I have a feed forward neural network trained with trainlm function.Now i want to change the learning rate and momentum.Is there a specific default range for learning rate or Do I need to find the optimum range for learning rate?

채택된 답변

GW
GW 2018년 11월 2일
편집: GW 2018년 11월 2일
use
options = trainingOptions('sgdm',...
'MaxEpochs',60, ...
'LearnRateDropFactor',0.1,...
'LearnRateDropPeriod',1,...
'LearnRateSchedule','piecewise',...
'ValidationData',{valImages,valLabels},...
'ValidationFrequency',50,...
'Momentum',0.85,...
'ValidationPatience',5,...
'InitialLearnRate',0.1,...
'MiniBatchSize',128,...
'Verbose',true,...
'executionenvironment','cpu',...
'Shuffle','every-epoch',...
'Plots','training-progress');
net = trainNetwork(...,...,...,options);
and choose the parameters accordingly.
For default values, see the docs of trainingOptions.
In terms of the learning rate and momentum, I typically start with a large one just to test the general behaviour and then I drop the learning rate to get more accurate results. Its difficult to give you some concrete advice because the optimal learning rate depends on your specific problem.
  댓글 수: 3
Pendar  Arzandeh
Pendar Arzandeh 2022년 5월 25일
Could you explain about trainlm? How to change learning rate? matlab allow changing mu and Gradient
Thanks
Katy
Katy 2023년 7월 19일
Is there a solution for this? I am also trying to change the learning rate with 'train' and 'trainlm' functions. Thanks!

댓글을 달려면 로그인하십시오.

추가 답변 (0개)

태그

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by