How to use Classification layer for dlnetwork

4 views (last 30 days)
I set up a custom training loop , and when i declare the layers i want to use, i cannot include Classificatio layer, otherwise i receive an erro which says "dlnework must not ha a return", i have to classify images so i need of that layer, so what should i do?

Answers (1)

Prince Kumar
Prince Kumar on 30 Mar 2022
To implement a custom training loop for your network, first convert it to a dlnetwork object. Do not include output layers in a dlnetwork object. Instead, you must specify the loss function in the custom training loop.
Please refer the following documentation for more information:


Find more on Deep Learning Toolbox 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!