Change Last Layer of Actor Network
3 views (last 30 days)
Show older comments
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 Comments
Answers (1)
Poorna
on 1 Sep 2023
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 Comments
See Also
Categories
Find more on Sequence and Numeric Feature Data Workflows 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!