def get_action_discrete(self, obs):
    mean = self.actor(obs)
    dist = Categorical(mean)
    action = dist.sample()
    log_prob = dist.log_prob(action)
    return (action.detach().numpy().item(), log_prob.detach().item())
def get_action_discrete(self, obs):
    mean = self.actor(obs)
    dist = Categorical(mean)
    action = dist.sample()
    log_prob = dist.log_prob(action)
    return (action.detach().numpy().item(), log_prob.detach().item())