Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: pi05
 pipeline_tag: robotics
 tags:
-- robotics
-- lerobot
 - pi05
+- lerobot
+- robotics
 ---
 
 # 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/task30/train/checkpoints/last/pretrained_model",
+  "base_model_name_or_path": "/data/taojiachen/Continuous_VLA/exp/sequential/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:0e43064f61aaa65aa23013f153ac3f6b5941483dffe39ffb53cdbad1bc7c8094
+oid sha256:720d80fd50305b29ba054075af898bb09ceb7fee63817252058099f88023c90c
 size 5161624

config.json

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

train_config.json

@@ -4,28 +4,22 @@
         "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,
@@ -49,7 +43,6 @@
             1511,
             1551,
             1591,
-            1630,
             1631,
             1671
         ],
@@ -173,7 +166,7 @@
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
+        "pretrained_path": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task30/train/checkpoints/last/pretrained_model",
         "paligemma_variant": "gemma_2b",
         "action_expert_variant": "gemma_300m",
         "dtype": "bfloat16",
@@ -217,7 +210,7 @@
         "scheduler_decay_steps": 30000,
         "scheduler_decay_lr": 2.5e-06
     },
-    "output_dir": "/data/taojiachen/Continuous_VLA/exp/er/spatial/seed42/task31/train",
+    "output_dir": "/data/taojiachen/Continuous_VLA/exp/sequential/spatial/seed42/task31/train",
     "job_name": "pi05",
     "resume": false,
     "seed": 42,