alexandersoare
commited on
Commit
•
0f3767e
1
Parent(s):
af403dc
update to align with 265b0ec44d42ea8cb912534df5b1e818d6caec0d
Browse files- config.yaml +7 -4
config.yaml
CHANGED
@@ -61,13 +61,16 @@ fps: 10
|
|
61 |
env:
|
62 |
name: pusht
|
63 |
task: PushT-v0
|
64 |
-
from_pixels: true
|
65 |
-
pixels_only: false
|
66 |
image_size: 96
|
67 |
-
episode_length: 300
|
68 |
-
fps: ${fps}
|
69 |
state_dim: 2
|
70 |
action_dim: 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
71 |
override_dataset_stats:
|
72 |
observation.image:
|
73 |
mean:
|
|
|
61 |
env:
|
62 |
name: pusht
|
63 |
task: PushT-v0
|
|
|
|
|
64 |
image_size: 96
|
|
|
|
|
65 |
state_dim: 2
|
66 |
action_dim: 2
|
67 |
+
fps: ${fps}
|
68 |
+
episode_length: 300
|
69 |
+
gym:
|
70 |
+
obs_type: pixels_agent_pos
|
71 |
+
render_mode: rgb_array
|
72 |
+
visualization_width: 384
|
73 |
+
visualization_height: 384
|
74 |
override_dataset_stats:
|
75 |
observation.image:
|
76 |
mean:
|