wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add end-to-end pusht eval test

This commit is contained in:
Simon Alibert 2024-03-09 16:37:34 +01:00
parent fa7a947acc
commit c6fb40fb29
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
.github/workflows/test.yml vendored
View File

@ -101,19 +101,22 @@ jobs:
hydra.job.name=pusht \ hydra.job.name=pusht \
env=pusht \ env=pusht \
wandb.enable=False \ wandb.enable=False \
offline_steps=1 \ offline_steps=2 \
online_steps=0 \ online_steps=0 \
device=cpu device=cpu \
# TODO(rcadene, aliberts): Add end-to-end test of eval checkpoint post training save_model=true \
# - name: Test eval pusht end-to-end save_freq=1 \
# run: | hydra.run.dir=tests/outputs/
# source .venv/bin/activate - name: Test eval pusht end-to-end
# python lerobot/scripts/eval.py run: |
# hydra.job.name=pusht \ source .venv/bin/activate
# env=pusht \ python lerobot/scripts/eval.py
# wandb.enable=False \ hydra.job.name=pusht \
# eval_episodes=1 \ env=pusht \
# device=cpu wandb.enable=False \
eval_episodes=1 \
device=cpu \
policy.pretrained_model_path=tests/outputs/models/1.pt
#---------------------------------------------- #----------------------------------------------
# cleanup # cleanup
#---------------------------------------------- #----------------------------------------------