I have a doubt for FC layer of my resnet. I don`t know if is it right.
Can you help me?
Maybe do i open a new topic?
    def __init__(self, n_input, n_output, n_hidden1, n_hidden2):
        super(ActorCriticModel, self).__init__()
        self.model = resnet50(pretrained=True)
        out = self.model.fc.in_features
        self.model.fc = Identity() #identity is a class where the forward return x element
        self.fcN = nn.Linear(out, n_hidden1)
        self.fcV = nn.Linear(out, n_hidden2)
        #self.model.fc = torch.cat(self.fcN + self.fcV)
        self.muX = nn.Linear(n_hidden1, 1)
        self.muZ = nn.Linear(n_hidden1, 1)
        self.sigma = nn.Linear(n_hidden1, 1)
        self.muO1 = nn.Linear(n_hidden1, 1)
        self.muO2 = nn.Linear(n_hidden1, 1)
        self.sigmaO = nn.Linear(n_hidden1, 1)
        self.value = nn.Linear(n_hidden2, 1)
        self.distributionXZ = torch.distributions.MultivariateNormal
        self.distributionO = torch.distributions.MultivariateNormal