squash commits

2024-09-11 21:09:17 -04:00
parent 8ce0aa1485
commit 2ddf63b8f5
200 changed files with 1240 additions and 1186 deletions
@@ -54,9 +54,7 @@ model:
  horizon_steps: ${horizon_steps}
  obs_dim: ${obs_dim}
  action_dim: ${action_dim}
-  transition_dim: ${transition_dim}
  denoising_steps: ${denoising_steps}
-  cond_steps: ${cond_steps}
  device: ${device}

 ema:
@@ -47,7 +47,7 @@ model:
    residual_style: False
    fixed_std: 0.1
    num_modes: ${num_modes}
-    cond_dim: ${obs_dim}
+    cond_dim: ${eval:'${obs_dim} * ${cond_steps}'}
    horizon_steps: ${horizon_steps}
    transition_dim: ${transition_dim}
  horizon_steps: ${horizon_steps}