How to choose the optimum values of the training options Transfer learning of the pretrain Deep networks in MATLAB

5 次查看(过去 30 天)
Dear all I have a question, I am using Transfer Learning of the pretraining Deep learning networks I would like to know how to choose the optimum values of the training options(Solver, initial learning rate, max epoch, miniBatchsize etc) should I keep trying different values or choosing the default values, please

回答(1 个)

Meet
Meet 2025-1-17
Hi Hussain,
To choose optimal training options for transfer learning, start with default settings as a baseline. Use 'adam' or 'sgdm' as solvers, with an initial learning rate around 1e-4 or 1e-3. Set 'maxEpochs' between 5-20, and try mini-batch sizes like 32 or 64. Adjust parameters based on validation performance, employing techniques like learning rate schedules and early stopping to refine results. Experiment with different such parameters to guide your choices.
Hope this clears the doubt!!

类别

Help CenterFile Exchange 中查找有关 Deep Learning Toolbox 的更多信息

产品


版本

R2021a

Community Treasure Hunt

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

Start Hunting!

Translated by