amend

pytorch · Jul 10, 2024 · 4e0596c · 4e0596c
1 parent 971c2ed
commit 4e0596c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sota-implementations/dreamer/config.yaml b/sota-implementations/dreamer/config.yaml
@@ -9,7 +9,7 @@ env:
   image_size : 64
   horizon: 500
   n_parallel_envs: 8
-  device:
+  device: cpu
 
 collector:
   total_frames: 5_000_000