How to use a userdefined algorithm for training DNN or within trainNetwork?

How to use a userdefined algorithm for training DNN or within trainingOptions?
Currently, it seems that only three options can only be available: "TrainingOptionsSGDM | TrainingOptionsRMSProp | TrainingOptionsADAM".
Thank you.

回答(1 个)

Hi Shashi Kant,
I understand that you are trying to use a custom algorithm for training a deep neural network.
You can create a custom training loop using “dlnetwork” object that allows automatic differentiation. Please refer to the below documentation to learn more about defining deep learning networks for custom training loops:
You can also follow the workflow shown in the below example to train a deep neural network using custom algorithm:
Hope it helps.
Regards,
Sai Pavan

类别

帮助中心File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

产品

版本

R2022a

Community Treasure Hunt

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

Start Hunting!

Translated by