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.

Answer 1

[Sai Pavan]

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:

https://www.mathworks.com/help/deeplearning/ug/define-custom-training-loops-loss-functions-and-networks.html

You can also follow the workflow shown in the below example to train a deep neural network using custom algorithm:

https://www.mathworks.com/help/deeplearning/ug/train-network-using-custom-training-loop.html

Hope it helps.

Regards,

Sai Pavan