61eab94af71466aaf1688c9aa6cd8c4600c291e4,ml/rl/training/dqn_trainer.py,DQNTrainer,get_detached_q_values,#DQNTrainer#Any#,198

Before Change

    def get_detached_q_values(self, states) -> Tuple[torch.Tensor, torch.Tensor]:
        with torch.no_grad():
            q_values = self.q_network(states)
            q_values_target = self.q_network_target(states)
        return q_values, q_values_target

    def get_next_action_q_values(self, states, next_actions):

After Change

         Gets the q values from the model and target networks 
        with torch.no_grad():
            input = rlt.StateInput(state=state)
            q_values = self.q_network(input).q_values
            q_values_target = self.q_network_target(input).q_values
        return q_values, q_values_target

    def train(self, training_batch):

Instances