How to change learning rate and momentum in 'trainlm' neural network?
7 visualizaciones (últimos 30 días)
Mostrar comentarios más antiguos
Avanthi Saumyamala
el 2 de Nov. de 2018
Comentada: Katy
el 19 de Jul. de 2023
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?
0 comentarios
Respuesta aceptada
GW
el 2 de Nov. de 2018
Editada: GW
el 2 de Nov. de 2018
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 comentarios
Pendar Arzandeh
el 25 de Mayo de 2022
Could you explain about trainlm? How to change learning rate? matlab allow changing mu and Gradient
Thanks
Katy
el 19 de Jul. de 2023
Is there a solution for this? I am also trying to change the learning rate with 'train' and 'trainlm' functions. Thanks!
Más respuestas (0)
Ver también
Categorías
Más información sobre Define Shallow Neural Network Architectures en Help Center y File Exchange.
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!