Merge pull request #15 from Cadene/user/aliberts/2024_03_09_add_ci_eval_test
Add end-to-end pusht eval test
This commit is contained in:
commit
1cc621ec36
|
@ -101,21 +101,20 @@ 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 \
|
||||||
# cleanup
|
env.episode_length=8 \
|
||||||
#----------------------------------------------
|
device=cpu \
|
||||||
- name: Cleanup
|
policy.pretrained_model_path=tests/outputs/models/1.pt
|
||||||
run: rm -rf diffusion_policy data
|
|
||||||
|
|
Loading…
Reference in New Issue