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()