fix

pytorch · Jun 13, 2023 · ed57fca · ed57fca
1 parent 7c48bff
commit ed57fca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intermediate_source/coding_ddpg.py b/intermediate_source/coding_ddpg.py
@@ -768,7 +768,7 @@ def make_ddpg_actor(
     ).to(device)
 
     # initialize lazy modules
-    qnet(actor(proof_environment.reset()))
+    qnet(actor(proof_environment.reset().to(device)))
     return actor, qnet