How to use Classification layer for dlnetwork

Question

MICHELE RUSSO 2022-3-21

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1676469-how-to-use-classification-layer-for-dlnetwork

回答： Prince Kumar 2022-3-30

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?

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Prince Kumar 2022-3-30

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1676469-how-to-use-classification-layer-for-dlnetwork#answer_930399

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:

https://www.mathworks.com/help/deeplearning/ref/dlnetwork.html