Change Last Layer of Actor Network
1 回表示 (過去 30 日間)
古いコメントを表示
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?
0 件のコメント
回答 (1 件)
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!
0 件のコメント
参考
カテゴリ
Help Center および File Exchange で Deep Learning Toolbox についてさらに検索
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!