Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: pi05
 pipeline_tag: robotics
 tags:
-- pi05
 - robotics
 - lerobot
+- pi05
 ---
 
 # Model Card for pi05

adapter_config.json

@@ -6,7 +6,7 @@
     "base_model_class": "PI05Policy",
     "parent_library": "lerobot.policies.pi05.modeling_pi05"
   },
-  "base_model_name_or_path": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
+  "base_model_name_or_path": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
   "bias": "none",
   "corda_config": null,
   "ensure_weight_tying": false,

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d4dcf6dc8f5e2b714c37c6b12d67f43fa864f37dd170438de6568047a0c288b9
+oid sha256:d1f994cbf612f5fd1494193c0d0632f344ca5115c6431ae3f0e8b6e9c5b73212
 size 12547808

config.json

@@ -41,7 +41,7 @@
     "private": null,
     "tags": null,
     "license": null,
-    "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
+    "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
     "paligemma_variant": "gemma_2b",
     "action_expert_variant": "gemma_300m",
     "dtype": "bfloat16",

train_config.json

@@ -4,22 +4,28 @@
         "root": null,
         "episodes": [
             31,
+            70,
             71,
             111,
             151,
             191,
             231,
             271,
+            310,
             311,
+            350,
             351,
             391,
             431,
             471,
             511,
             551,
+            590,
             591,
+            630,
             631,
             671,
+            710,
             711,
             751,
             791,
@@ -43,6 +49,7 @@
             1511,
             1551,
             1591,
+            1630,
             1631,
             1671
         ],
@@ -166,7 +173,7 @@
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
+        "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
         "paligemma_variant": "gemma_2b",
         "action_expert_variant": "gemma_300m",
         "dtype": "bfloat16",
@@ -210,7 +217,7 @@
         "scheduler_decay_steps": 30000,
         "scheduler_decay_lr": 2.5e-06
     },
-    "output_dir": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task31/train",
+    "output_dir": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task31/train",
     "job_name": "pi05",
     "resume": false,
     "seed": 42,