Save intermediate model in matlab while training a deep learning model and resume training from that stage later
11 views (last 30 days)
Show older comments
parvathy prathap
on 12 Oct 2021
Answered: Mahesh Taparia
on 15 Oct 2021
Hi,
I am training a 3 pipeline deep learning model in matlab which takes a lot of time to train. I need to store intermediate variable values while training, stop the training process and then resume training at a later point from the stage at which the training was stopped previously. Does Matlab have any options to do this? Any help in this regard would be highly appreciated.
Thanks in Advance
0 Comments
Accepted Answer
Mahesh Taparia
on 15 Oct 2021
Hi
You can set the checkpoint path in trainingOptions as suggested in the above answer. The trained weights will be saved into the specified path as a mat file. To resume the training process, you can load those weights in the net variable and start the training process. For example, you can refer this documentation.
Hope it will help!
0 Comments
More Answers (1)
yanqi liu
on 15 Oct 2021
sir,may be use CheckpointPath,such as
options = trainingOptions('sgdm', ...
'MaxEpochs', 5, ...
'MiniBatchSize', 1, ...
'InitialLearnRate', 1e-3, ...
'CheckpointPath', tempdir);
0 Comments
See Also
Categories
Find more on Deep Learning with Images in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!