diff --git a/sota-implementations/dreamer/config.yaml b/sota-implementations/dreamer/config.yaml index 12e7c4e6446..604e1ac546a 100644 --- 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