Skip to content

Commit ed57fca

Browse files
committed
fix
1 parent 7c48bff commit ed57fca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/coding_ddpg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -768,7 +768,7 @@ def make_ddpg_actor(
768768
).to(device)
769769

770770
# initialize lazy modules
771-
qnet(actor(proof_environment.reset()))
771+
qnet(actor(proof_environment.reset().to(device)))
772772
return actor, qnet
773773

774774

0 commit comments

Comments
 (0)