Change Last Layer of Actor Network

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?

Answer 1

[Poorna]

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!