def get_actions(self, observations):
    guessed_actions = self.actor_model(torch.Tensor(observations).to(self.actor_model.device))
    return guessed_actions
def get_actions(self, observations):
    guessed_actions = self.actor_model(torch.Tensor(observations).to(self.actor_model.device))
    return guessed_actions