def cuda(self): Agent.cuda(self) self.e_DQN = self.e_DQN.cuda() self.t_DQN = self.t_DQN.cuda()
def cuda(self): Agent.cuda(self) self.e_NAF = self.e_NAF.cuda() self.t_NAF = self.t_NAF.cuda()
def cuda(self): Agent.cuda(self) self.policy = self.policy.cuda() if self.value_type is not None: self.value = self.value.cuda()
def cuda(self): Agent.cuda(self) self.e_Actor = self.e_Actor.cuda() self.e_Critic = self.e_Critic.cuda() self.t_Actor = self.t_Actor.cuda() self.t_Critic = self.t_Critic.cuda()