Robotics
Transformers
Safetensors
Inference Endpoints
alexandersoare commited on
Commit
0f3767e
1 Parent(s): af403dc

update to align with 265b0ec44d42ea8cb912534df5b1e818d6caec0d

Browse files
Files changed (1) hide show
  1. 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: