MichaelYu781 commited on
Commit
4cc5408
·
verified ·
1 Parent(s): 07713e8

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +2 -0
config.yaml CHANGED
@@ -26,6 +26,8 @@ framework:
26
  action_model_type: DiT-B
27
  action_hidden_dim: 768
28
  action_dim: 7
 
 
29
  use_ema: false
30
  future_action_window_size: 7
31
  past_action_window_size: 0
 
26
  action_model_type: DiT-B
27
  action_hidden_dim: 768
28
  action_dim: 7
29
+ input_dim: 2048
30
+ ouptput_dim: 768
31
  use_ema: false
32
  future_action_window_size: 7
33
  past_action_window_size: 0