lerobot icon indicating copy to clipboard operation
lerobot copied to clipboard

Add RLBench

Open eugeneteoh opened this issue 8 months ago • 1 comments

What this does

Add RLBench as a benchmark. https://github.com/stepjam/RLBench

How it was tested

How to checkout & try? (for the reviewer)

conda create -y -n lerobot python=3.10 && conda activate lerobot
conda env config vars set COPPELIASIM_ROOT=/path/to/coppeliasim
conda env config vars set LD_LIBRARTY_PATH=$LD_LIBRARY_PATH:$COPPELIASIM_ROOT QT_QPA_PLATFORM_PLUGIN_PATH=$COPPELIASIM_ROOT 
conda activate lerobot
pip install -e ".[rlbench]"
# First, generate raw rlbench data
rlbench-generate-data --save_path=data/raw/rlbench --tasks=reach_target --episodes_per_task=100 --variations=1 --processes=4
# Push dataset to huggingface hub
python lerobot/scripts/push_dataset_to_hub.py --raw-dir data/raw/rlbench/open_drawer/variation0 --raw-format rlbench --repo-id eugeneteoh/rlbench_reach_target_variation_0 --local-dir data/eugeneteoh/rlbench_open_drawer_variation_0 --push-to-hub=1
# Run training
python lerobot/scripts/train.py policy=act_rlbench env=rlbench

eugeneteoh avatar Jun 23 '24 14:06 eugeneteoh