tests for tdmpc and diffusion policy are passing

2024-04-09 02:50:32 +00:00
parent 1e09507bc1
commit 73dfa3c8e3
6 changed files with 19 additions and 17 deletions
--- a/lerobot/configs/policy/diffusion.yaml
+++ b/lerobot/configs/policy/diffusion.yaml
@@ -38,6 +38,7 @@ policy:

  horizon: ${horizon}
  n_obs_steps: ${n_obs_steps}
+  n_action_steps: ${n_action_steps}
  num_inference_steps: 100
  obs_as_global_cond: ${obs_as_global_cond}
  # crop_shape: null
--- a/lerobot/configs/policy/tdmpc.yaml
+++ b/lerobot/configs/policy/tdmpc.yaml
@@ -36,7 +36,6 @@ policy:
  log_std_max: 2

  # learning
-  batch_size: 256
  max_buffer_size: 10000
  horizon: 5
  reward_coef: 0.5