Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: pi05
 pipeline_tag: robotics
 tags:
-- lerobot
-- robotics
 - pi05
+- robotics
+- lerobot
 ---
 
 # 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/er/spatial/seed42/task31/train/checkpoints/last/pretrained_model",
+  "base_model_name_or_path": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task31/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:bac9ac47d9adb81c1a0331e9e0c46c523b9d3afb10f21a8c23195610fa52ff56
+oid sha256:d05538b5ff8064b42f85403d54a1a0a087464ad366a5ca12a423bdb9b99e5b4e
 size 12547808

config.json

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

train_config.json

@@ -4,32 +4,22 @@
         "root": null,
         "episodes": [
             32,
-            70,
-            71,
             72,
-            111,
             112,
             152,
             192,
-            231,
             232,
-            271,
             272,
-            310,
             312,
-            350,
             352,
             392,
             432,
             472,
             512,
             552,
-            590,
             592,
-            630,
             632,
             672,
-            710,
             712,
             752,
             792,
@@ -40,7 +30,6 @@
             992,
             1032,
             1072,
-            1111,
             1112,
             1152,
             1192,
@@ -48,15 +37,12 @@
             1272,
             1312,
             1352,
-            1391,
             1392,
             1432,
             1472,
-            1511,
             1512,
             1552,
             1592,
-            1630,
             1632,
             1672
         ],
@@ -180,7 +166,7 @@
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task31/train/checkpoints/last/pretrained_model",
+        "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task31/train/checkpoints/last/pretrained_model",
         "paligemma_variant": "gemma_2b",
         "action_expert_variant": "gemma_300m",
         "dtype": "bfloat16",
@@ -224,7 +210,7 @@
         "scheduler_decay_steps": 30000,
         "scheduler_decay_lr": 2.5e-06
     },
-    "output_dir": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task32/train",
+    "output_dir": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task32/train",
     "job_name": "pi05",
     "resume": false,
     "seed": 42,