shgh:

        self.linear1 = nn.Linear(state_size, hidden_size)



change the state_size to 1 rather than 3 will make it pass through that layer. If you’re sure the state_size is 3, then the issue lies with state instead