Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)

2024-02-26 01:10:09 +00:00
parent 5a219fed6e
commit 21670dce90
12 changed files with 306 additions and 443 deletions
--- a/lerobot/configs/policy/tdmpc.yaml
+++ b/lerobot/configs/policy/tdmpc.yaml
@@ -5,8 +5,6 @@ policy:

  reward_scale: 1.0

-  # xarm_lift
-  train_steps: ${train_steps}
  episode_length: ${env.episode_length}
  discount: 0.9
  modality: 'all'