Overwrite ACT stack_blocks_v2 qpos trained on stack_blocks_v2_fixed (leading static frames trimmed)

config.json

@@ -0,0 +1,72 @@
+{
+  "action_dim": 7,
+  "architectures": [
+    "ACTPolicy"
+  ],
+  "args": {
+    "action_dim": 7,
+    "backbone": "resnet18",
+    "camera_names": [
+      "primary",
+      "wrist"
+    ],
+    "chunk_size": 16,
+    "dec_layers": 7,
+    "dilation": false,
+    "dim_feedforward": 3200,
+    "dropout": 0.1,
+    "enc_layers": 4,
+    "hidden_dim": 512,
+    "history_len": 1,
+    "image_size": [
+      256,
+      256
+    ],
+    "kl_weight": 10,
+    "lr_backbone": 1e-05,
+    "masks": false,
+    "nheads": 8,
+    "num_queries": 16,
+    "position_embedding": "sine",
+    "pre_norm": false,
+    "prediction_len": 16,
+    "raw_size": [
+      640,
+      480
+    ],
+    "state_dim": 7
+  },
+  "backbone": "resnet18",
+  "camera_names": [
+    "primary",
+    "wrist"
+  ],
+  "chunk_size": 16,
+  "dec_layers": 7,
+  "dilation": false,
+  "dim_feedforward": 3200,
+  "dropout": 0.1,
+  "enc_layers": 4,
+  "hidden_dim": 512,
+  "history_len": 1,
+  "image_size": [
+    256,
+    256
+  ],
+  "kl_weight": 10,
+  "lr_backbone": 1e-05,
+  "masks": false,
+  "nheads": 8,
+  "num_queries": 16,
+  "position_embedding": "sine",
+  "pre_norm": false,
+  "prediction_len": 16,
+  "raw_size": [
+    640,
+    480
+  ],
+  "state_dim": 7,
+  "torch_dtype": "float32",
+  "transformers_version": "4.45.2",
+  "type": "policy.act"
+}

example_data/action_raw.csv

@@ -0,0 +1,17 @@
+action_0,action_1,action_2,action_3,action_4,action_5,action_6
+-3.025104664266109467e-02,1.033142805099487305e+00,-1.418284893035888672e+00,-7.773706316947937012e-01,1.956540107727050781e+00,5.858902931213378906e-01,6.953705549240112305e-01
+-3.580585494637489319e-02,1.035427093505859375e+00,-1.413290262222290039e+00,-7.762162089347839355e-01,1.957386493682861328e+00,5.856456160545349121e-01,6.953705549240112305e-01
+-6.156156957149505615e-02,1.041838049888610840e+00,-1.396861553192138672e+00,-7.709288597106933594e-01,1.961378812789916992e+00,5.871674418449401855e-01,6.953705549240112305e-01
+-9.063446521759033203e-02,1.043805241584777832e+00,-1.387123584747314453e+00,-7.600581645965576172e-01,1.964160680770874023e+00,5.944994688034057617e-01,6.953705549240112305e-01
+-1.128363236784934998e-01,1.046252012252807617e+00,-1.376459121704101562e+00,-7.487570047378540039e-01,1.967370510101318359e+00,6.017965078353881836e-01,6.953705549240112305e-01
+-1.357023864984512329e-01,1.047803282737731934e+00,-1.367038965225219727e+00,-7.363485693931579590e-01,1.969274163246154785e+00,6.097449660301208496e-01,6.953705549240112305e-01
+-1.526241600513458252e-01,1.049885034561157227e+00,-1.357456445693969727e+00,-7.245928049087524414e-01,1.971589922904968262e+00,6.171205043792724609e-01,6.953705549240112305e-01
+-1.644703149795532227e-01,1.052649259567260742e+00,-1.347732424736022949e+00,-7.126069068908691406e-01,1.974536776542663574e+00,6.247575283050537109e-01,6.953705549240112305e-01
+-1.745702624320983887e-01,1.055898070335388184e+00,-1.337904930114746094e+00,-7.011149525642395020e-01,1.977974653244018555e+00,6.324075460433959961e-01,6.953705549240112305e-01
+-1.942932456731796265e-01,1.064398288726806641e+00,-1.313560247421264648e+00,-6.784186959266662598e-01,1.988250851631164551e+00,6.493363380432128906e-01,6.953705549240112305e-01
+-2.088605016469955444e-01,1.069393038749694824e+00,-1.296988606452941895e+00,-6.668682098388671875e-01,1.995654106140136719e+00,6.596608161926269531e-01,6.953705549240112305e-01
+-2.328763306140899658e-01,1.072506070137023926e+00,-1.276248097419738770e+00,-6.497988700866699219e-01,2.005367755889892578e+00,6.783718466758728027e-01,6.953705549240112305e-01
+-2.492536902427673340e-01,1.074169158935546875e+00,-1.260091900825500488e+00,-6.323590874671936035e-01,2.013840913772583008e+00,6.979106664657592773e-01,6.953705549240112305e-01
+-2.673997282981872559e-01,1.076789140701293945e+00,-1.242738842964172363e+00,-6.173431873321533203e-01,2.023182630538940430e+00,7.159383296966552734e-01,6.953705549240112305e-01
+-2.928298413753509521e-01,1.077711343765258789e+00,-1.228341937065124512e+00,-6.018940806388854980e-01,2.033122301101684570e+00,7.386404871940612793e-01,6.953705549240112305e-01
+-3.252362906932830811e-01,1.082095623016357422e+00,-1.203530550003051758e+00,-5.735341310501098633e-01,2.054781913757324219e+00,7.834547162055969238e-01,6.953705549240112305e-01

example_data/info.txt

@@ -0,0 +1,26 @@
+=== Training Example Data Info ===
+
+This example is saved from the raw training dataset (before data_processor).
+Data is saved in unnormalized form to match testing phase format.
+
+Sample keys:
+  image: shape=torch.Size([2, 3, 256, 256]), dtype=torch.uint8
+  state: shape=torch.Size([7]), dtype=torch.float32
+  action: shape=torch.Size([16, 7]), dtype=torch.float32
+  is_pad: shape=torch.Size([16]), dtype=torch.bool
+  raw_lang: str
+  reasoning: dict
+  timestamp: int
+  episode_id: int
+  action_raw: shape=torch.Size([16, 7]), dtype=torch.float32
+  dataset_id: str
+
+Files saved:
+  - camera_{i}.png: raw images from dataset (unnormalized, original resolution)
+  - state_raw.csv: raw state values (unnormalized)
+  - action_raw.csv: raw action values (unnormalized)
+  - raw_lang.txt: language instruction (if available)
+  - reasoning.json: reasoning data (if available)
+  - info.txt: this file
+
+Note: These raw values can be directly compared with testing phase examples.

example_data/raw_lang.txt

@@ -0,0 +1 @@
+stack_blocks_v2

example_data/state_raw.csv

@@ -0,0 +1,2 @@
+state_0,state_1,state_2,state_3,state_4,state_5,state_6
+-3.025104664266109467e-02,1.033142805099487305e+00,-1.418284893035888672e+00,-7.773706316947937012e-01,1.956540107727050781e+00,5.858902931213378906e-01,6.953705549240112305e-01

log/events.out.tfevents.1784548443.asi-keepalive-4a-202607152040--b75f27509fb0-do5xieuiqx.3960200.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fb61259ef7debf10585f9106b544a47ebe0223c35d2e5b24988d55ea504775db
+size 644264

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4796f9de36f58e9f6432805851fb8577c3698426601628da1062f007d5c03255
+size 335617008

normalize.json

@@ -0,0 +1,16 @@
+{
+  "version": "2.0",
+  "datasets": [
+    {
+      "dataset_id": "stack_blocks_v2_qpos",
+      "ctrl_space": "joint",
+      "ctrl_type": "abs"
+    }
+  ],
+  "state": {
+    "stack_blocks_v2_qpos": "zscore"
+  },
+  "action": {
+    "stack_blocks_v2_qpos": "zscore"
+  }
+}

policy_metadata.json

@@ -0,0 +1,12 @@
+{
+  "policy_module": "policy.act",
+  "policy_name": "act_cam2_7dof",
+  "raw_size": [
+    640,
+    480
+  ],
+  "image_size": [
+    256,
+    256
+  ]
+}

stack_blocks_v2_qpos_stats_joint_abs.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff0b4ab875bf201a879a1ffb3b4ed674d69ad178885142856141f951936881f9
+size 1379

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c24a632dba01c079c3ea8dcb13496b85e242464a891e90356654e57096103e31
+size 5649