lerobot/lerobot/configs/env/aloha.yaml

26 lines
390 B
YAML

# @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: [3, 480, 640]
action_repeat: 1
episode_length: 400
fps: ${fps}
policy:
state_dim: 14
action_dim: 14