This commit is contained in:
Simon Alibert 2024-03-05 16:26:31 +01:00
parent 7a20ef65f6
commit 12af67066d
1 changed files with 17 additions and 16 deletions

View File

@ -68,24 +68,25 @@ jobs:
#---------------------------------------------- #----------------------------------------------
# run tests # run tests
#---------------------------------------------- #----------------------------------------------
- name: Activate venv - name: Test train pusht end-to-end
run: | run: |
source .venv/bin/activate source .venv/bin/activate
- name: Test train pusht end-to-end python lerobot/scripts/train.py \
run: python lerobot/scripts/train.py \ hydra.job.name=pusht \
hydra.job.name=pusht \ env=pusht \
env=pusht \ wandb.enable=False \
wandb.enable=False \ offline_steps=1 \
offline_steps=1 \ online_steps=1 \
online_steps=1 \ device=cpu
device=cpu # - name: Test eval pusht end-to-end
# - name: Test eval pusht # run: |
# run: python lerobot/scripts/eval.py # source .venv/bin/activate
# hydra.job.name=pusht \ # python lerobot/scripts/eval.py
# env=pusht \ # hydra.job.name=pusht \
# wandb.enable=False \ # env=pusht \
# eval_episodes=1 \ # wandb.enable=False \
# device=cpu # eval_episodes=1 \
# device=cpu
#---------------------------------------------- #----------------------------------------------
# cleanup # cleanup
#---------------------------------------------- #----------------------------------------------