Skip to content

Commit

Permalink
Reactivate tdmpc online integration test
Browse files Browse the repository at this point in the history
  • Loading branch information
aliberts committed Jan 24, 2025
1 parent cf777e9 commit 3cfbb20
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -131,13 +131,12 @@ test-tdmpc-ete-eval:
--eval.batch_size=1 \
--device=$(DEVICE)

# FIXME: currently broken
# test-tdmpc-ete-train-with-online:
# python lerobot/scripts/train.py \
test-tdmpc-ete-train-with-online:
python lerobot/scripts/train.py \
--policy.type=tdmpc \
--env.type=pusht \
--env.obs_type=environment_state_agent_pos \
--env.episode_length=10 \
--env.episode_length=5 \
--dataset.repo_id=lerobot/pusht_keypoints \
--dataset.image_transforms.enable=true \
--dataset.episodes='[0]' \
Expand All @@ -147,7 +146,7 @@ test-tdmpc-ete-eval:
--online.rollout_n_episodes=2 \
--online.rollout_batch_size=2 \
--online.steps_between_rollouts=10 \
--online.buffer_capacity=15 \
--online.buffer_capacity=1000 \
--online.env_seed=10000 \
--save_checkpoint=false \
--save_freq=10 \
Expand Down

0 comments on commit 3cfbb20

Please sign in to comment.