ISdept
/

transformer-diffusion

Model card Files Files and versions

ISdept commited on 15 days ago

Commit

9111220

·

verified ·

1 Parent(s): f94e075

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -63,7 +63,7 @@
         224
     ],
     "num_cameras": 3,
-    "vision_freeze_layers": 8,
     "cameras_for_vision_state_concat": [
         "observation.images.front"
     ],
@@ -72,7 +72,7 @@
     "d_model": 512,
     "nhead": 8,
     "num_encoder_layers": 4,
-    "num_decoder_layers": 4,
     "dim_feedforward": 512,
     "diffusion_step_embed_dim": 256,
     "kernel_size": 3,

         224
     ],
     "num_cameras": 3,
+    "vision_freeze_layers": 6,
     "cameras_for_vision_state_concat": [
         "observation.images.front"
     ],
     "d_model": 512,
     "nhead": 8,
     "num_encoder_layers": 4,
+    "num_decoder_layers": 8,
     "dim_feedforward": 512,
     "diffusion_step_embed_dim": 256,
     "kernel_size": 3,