I think the value of the **q_eval4next** need a new line of code to get, could you please give a tip how to modify it ? q_next, q_eval4next = self.forward(batch_memory[:, -self.n_features:], lstm_batch_memory[:, :, self.n_lstm_state:])