lerobot
lerobot copied to clipboard
⚡️add florence+linear action head policy, and adapt it to pusht env
What this does
add florence+linear action head policy, and adapt it to pusht env
How to checkout & try? (for the reviewer)
Examples:
pip install -e ".[florence]"
python lerobot/scripts/train.py policy=florence env=pusht dataset_repo_id=lerobot/pusht wandb.enable=True
TODO
- Tune the performance for pusht
- Whether it satisfy the coding format of LeRobot?
- adapt to
aloha-gymenv - add diffusion action head (Chi Cheng's diffusion transformer & DiT developed by @mbreuss at KIT)