Update config.json

config.json

@@ -41,7 +41,7 @@
     "device": "cuda",
     "use_amp": false,
     "push_to_hub": true,
-    "repo_id": "Temmp1e/pi0_grab",
+    "repo_id": "mizutoukotori/pi0_so101",
     "private": null,
     "tags": null,
     "license": null,
@@ -50,37 +50,28 @@
     "max_state_dim": 32,
     "max_action_dim": 32,
     "resize_imgs_with_padding": [
-        512,
-        512
+        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": true,
+    "train_expert_only": false,
     "train_state_proj": true,
-    "optimizer_lr": 0.0001,
+    "optimizer_lr": 2.5e-05,
     "optimizer_betas": [
         0.9,
         0.95
     ],
     "optimizer_eps": 1e-08,
     "optimizer_weight_decay": 1e-10,
-    "optimizer_grad_clip_norm": 10.0,
     "scheduler_warmup_steps": 1000,
     "scheduler_decay_steps": 30000,
-    "scheduler_decay_lr": 2.5e-06,
-    "add_image_special_tokens": false,
-    "attention_mode": "cross_attn",
-    "prefix_length": 0,
-    "pad_language_to": "max_length",
-    "num_expert_layers": 0,
-    "num_vlm_layers": 16,
-    "self_attn_every_n_layers": 2,
-    "expert_width_multiplier": 0.75,
-    "min_period": 0.004,
-    "max_period": 4.0
+    "scheduler_decay_lr": 2.5e-06
 }