Add aloha dataset

This commit is contained in:
Remi Cadene
2024-03-06 10:15:11 +00:00
parent 49c0955f97
commit d782b029e1
3 changed files with 216 additions and 0 deletions

25
lerobot/configs/env/aloha.yaml vendored Normal file
View File

@@ -0,0 +1,25 @@
# @package _global_
eval_episodes: 50
eval_freq: 7500
save_freq: 75000
log_freq: 250
# TODO: same as simxarm, need to adjust
offline_steps: 25000
online_steps: 25000
fps: 50
env:
name: aloha
task: sim_insertion_human
from_pixels: True
pixels_only: False
image_size: 96
action_repeat: 1
episode_length: 300
fps: ${fps}
policy:
state_dim: 2
action_dim: 2