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

9 views (last 30 days)
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

Answers (1)

Meet
Meet on 17 Jan 2025 at 12:39
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!!

Categories

Find more on Sequence and Numeric Feature Data Workflows in Help Center and File Exchange

Products


Release

R2021a

Community Treasure Hunt

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

Start Hunting!