Change Last Layer of Actor Network
1 vue (au cours des 30 derniers jours)
Afficher commentaires plus anciens
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 commentaires
Réponses (1)
Poorna
le 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 commentaires
Voir également
Catégories
En savoir plus sur Deep Learning Toolbox dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!