How to use Classification layer for dlnetwork

4 visualizzazioni (ultimi 30 giorni)
MICHELE RUSSO
MICHELE RUSSO il 21 Mar 2022
Risposto: Prince Kumar il 30 Mar 2022
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?

Risposte (1)

Prince Kumar
Prince Kumar il 30 Mar 2022
Hi,
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:

Community Treasure Hunt

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

Start Hunting!

Translated by