def forward(self, state):
    state_values = self.decoder(self.encoder(state))
    return state_values
def forward(self, state):
    state_values = self.decoder(self.encoder(state))
    return state_values