Skip to content

Commit dc46762

Browse files
committed
fix: set the default device to gpu for minimal working example
1 parent 41d79fb commit dc46762

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

repro/sec7_3_rl_train/impls/tempo_ppo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ def get_tempo_ppo_executor( # noqa: C901
183183
# NOTE: Default obs shape for trivial env
184184
"obs_shape": (3, 4, 4),
185185
"seed": 0,
186-
"dev": "fake-gpu",
186+
"dev": "gpu",
187187
"iterations": 50,
188188
# PPO hyperparams
189189
"gamma": 0.99,

0 commit comments

Comments
 (0)