removed unnecessary .cuda()

JunchenJin · May 18, 2018 · 00ae5cd · 00ae5cd
1 parent 7baf951
commit 00ae5cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TD3.py b/TD3.py
@@ -65,7 +65,7 @@ def forward(self, x, u):
 class TD3(object):
 	def __init__(self, state_dim, action_dim, max_action):
 		self.actor = Actor(state_dim, action_dim, max_action)
-		self.actor_target = Actor(state_dim, action_dim, max_action).cuda()
+		self.actor_target = Actor(state_dim, action_dim, max_action)
 		self.actor_target.load_state_dict(self.actor.state_dict())
 		self.actor_optimizer = torch.optim.Adam(self.actor.parameters())