WIP tdmpc

2024-04-05 13:40:31 +00:00
parent ab3cd3a7ba
commit f56b1a0e16
4 changed files with 121 additions and 94 deletions
--- a/lerobot/configs/env/simxarm.yaml
+++ b/lerobot/configs/env/simxarm.yaml
@@ -17,7 +17,7 @@ env:
  from_pixels: True
  pixels_only: False
  image_size: 84
-  action_repeat: 2
+  # action_repeat: 2  # we can remove if policy has n_action_steps=2
  episode_length: 25
  fps: ${fps}

--- a/lerobot/configs/policy/tdmpc.yaml
+++ b/lerobot/configs/policy/tdmpc.yaml
@@ -1,6 +1,6 @@
 # @package _global_

-n_action_steps: 1
+n_action_steps: 2
 n_obs_steps: 1

 policy: