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
