tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Enable tests for TD-MPC (#160)

Browse Source
This commit is contained in:
Alexander Soare
2024-05-09 13:42:12 +01:00
committed by GitHub
parent 7bb5b15f4c
commit e89521dfa0
7 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Makefile
View File

@@ -22,9 +22,8 @@ test-end-to-end:
${MAKE} test-act-ete-eval ${MAKE} test-act-ete-eval
${MAKE} test-diffusion-ete-train ${MAKE} test-diffusion-ete-train
${MAKE} test-diffusion-ete-eval ${MAKE} test-diffusion-ete-eval
# TODO(rcadene, alexander-soare): enable end-to-end tests for tdmpc ${MAKE} test-tdmpc-ete-train
# ${MAKE} test-tdmpc-ete-train ${MAKE} test-tdmpc-ete-eval
# ${MAKE} test-tdmpc-ete-eval
${MAKE} test-default-ete-eval ${MAKE} test-default-ete-eval
test-act-ete-train: test-act-ete-train:
@@ -80,7 +79,7 @@ test-tdmpc-ete-train:
policy=tdmpc \ policy=tdmpc \
env=xarm \ env=xarm \
env.task=XarmLift-v0 \ env.task=XarmLift-v0 \
dataset_repo_id=lerobot/xarm_lift_medium_replay \ dataset_repo_id=lerobot/xarm_lift_medium \
wandb.enable=False \ wandb.enable=False \
training.offline_steps=2 \ training.offline_steps=2 \
training.online_steps=2 \ training.online_steps=2 \

2
lerobot/configs/policy/tdmpc.yaml
View File

@@ -1,7 +1,7 @@
# @package _global_ # @package _global_
seed: 1 seed: 1
dataset_repo_id: lerobot/xarm_lift_medium_replay dataset_repo_id: lerobot/xarm_lift_medium
training: training:
offline_steps: 25000 offline_steps: 25000

BIN
tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/save_policy_to_safetensors/xarm_tdmpc/grad_stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/save_policy_to_safetensors/xarm_tdmpc/output_dict.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/save_policy_to_safetensors/xarm_tdmpc/param_stats.safetensors Normal file
View File

Binary file not shown.

2
tests/test_policies.py
View File

@@ -236,7 +236,7 @@ def test_normalize(insert_temporal_dim):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"env_name, policy_name, extra_overrides", "env_name, policy_name, extra_overrides",
[ [
# ("xarm", "tdmpc", ["policy.n_action_repeats=2"]), ("xarm", "tdmpc", []),
( (
"pusht", "pusht",
"diffusion", "diffusion",