ISdept
/

transformer-diffusion

Model card Files Files and versions

ISdept commited on 29 days ago

Commit

8c4e3c5

·

verified ·

1 Parent(s): f139d12

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
     "type": "transformer_diffusion",
-    "n_obs_steps": 2,
     "input_features": {
         "observation.state": {
             "type": "STATE",
@@ -69,7 +69,7 @@
     "action_dim": 7,
     "d_model": 512,
     "nhead": 8,
-    "num_encoder_layers": 3,
     "num_decoder_layers": 6,
     "dim_feedforward": 512,
     "diffusion_step_embed_dim": 256,

 {
     "type": "transformer_diffusion",
+    "n_obs_steps": 4,
     "input_features": {
         "observation.state": {
             "type": "STATE",
     "action_dim": 7,
     "d_model": 512,
     "nhead": 8,
+    "num_encoder_layers": 4,
     "num_decoder_layers": 6,
     "dim_feedforward": 512,
     "diffusion_step_embed_dim": 256,