def cuda(self):
    self.observations = self.observations.cuda()
    self.states = self.states.cuda()
    self.rewards = self.rewards.cuda()
    self.value_preds = self.value_preds.cuda()
    self.returns = self.returns.cuda()
    self.action_log_probs = self.action_log_probs.cuda()
    self.actions = self.actions.cuda()
    self.masks = self.masks.cuda()
def cuda(self):
    self.observations = self.observations.cuda()
    self.states = self.states.cuda()
    self.rewards = self.rewards.cuda()
    self.value_preds = self.value_preds.cuda()
    self.returns = self.returns.cuda()
    self.action_log_probs = self.action_log_probs.cuda()
    self.actions = self.actions.cuda()
    self.masks = self.masks.cuda()