Update config.json

config.json

@@ -9,64 +9,25 @@
     "input_features": {
         "observation.state": {
             "type": "STATE",
-            "shape": [
-                6
-            ]
+            "shape": [6]
         },
         "observation.images.laptop": {
             "type": "VISUAL",
-            "shape": [
-                3,
-                480,
-                640
-            ]
+            "shape": [3, 480, 640]
         },
         "observation.images.phone": {
             "type": "VISUAL",
-            "shape": [
-                3,
-                480,
-                640
-            ]
+            "shape": [3, 480, 640]
         }
     },
     "output_features": {
         "action": {
             "type": "ACTION",
-            "shape": [
-                6
-            ]
+            "shape": [6]
         }
     },
     "device": "cuda",
     "use_amp": false,
     "chunk_size": 50,
-    "n_action_steps": 50,
-    "max_state_dim": 32,
-    "max_action_dim": 32,
-    "resize_imgs_with_padding": [
-        224,
-        224
-    ],
-    "empty_cameras": 0,
-    "adapt_to_pi_aloha": false,
-    "use_delta_joint_actions_aloha": false,
-    "tokenizer_max_length": 48,
-    "proj_width": 1024,
-    "num_steps": 10,
-    "use_cache": true,
-    "attention_implementation": "eager",
-    "freeze_vision_encoder": true,
-    "train_expert_only": false,
-    "train_state_proj": true,
-    "optimizer_lr": 5e-05,
-    "optimizer_betas": [
-        0.9,
-        0.95
-    ],
-    "optimizer_eps": 1e-08,
-    "optimizer_weight_decay": 1e-10,
-    "scheduler_warmup_steps": 1000,
-    "scheduler_decay_steps": 30000,
-    "scheduler_decay_lr": 5e-06
+    "n_action_steps": 50
 }