Change Last Layer of Actor Network

2 次查看（过去 30 天）
Muhammed Murat Özbek
回答： Poorna 2023-9-1
Hi, i want to change last layer of actor network 3 to 4. i have pretrained network which last layer has 3 output and trained on same environment. But i need to change it to 4(adding new actions). my weights are in .mat format
How can i do it?

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

回答（1 个）

Poorna
Poorna 2023-9-1
Hi,
I understand that you would like to replace the output layer consisting of 3 nodes to 4 nodes. You can do this by using the “replaceLayer” function.
After loading the pretrained network, you could do the following.
actorNetwork = replaceLayer(actorNetwork, 'Output', fullyConnectedLayer(4, 'Name', 'newOutput'));
Here I assume that actorNetwork is the pretrained Network which is loaded from the weights.mat file.
Hope this Helps!

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

产品


版本

R2022a

Community Treasure Hunt

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

Start Hunting!

Translated by