Upload policy weights, train config and readme

README.md

@@ -1,13 +1,13 @@
 ---
-datasets: aic-cable-insertion-tele-v174_no_zero_actions_state_filtered
+datasets: kejia98/aic-cable-insertion-tele-v180-state-processed_state_filtered
 library_name: lerobot
 license: apache-2.0
 model_name: ditflow
 pipeline_tag: robotics
 tags:
-- robotics
-- ditflow
 - lerobot
+- ditflow
+- robotics
 ---
 
 # Model Card for ditflow

config.json

@@ -5,7 +5,7 @@
         "observation.state": {
             "type": "STATE",
             "shape": [
-                7
+                13
             ]
         },
         "observation.images.left_camera": {

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6cf7fddf2b474c0e9d9876eb82bc5aeb707c91f96654797b00c8e38d8ddc89ba
-size 257769776
+oid sha256:f311a81c250e6f985bf233a58ccbaf853a310405e88f627a6b1b792be2f04623
+size 257794352

train_config.json

@@ -1,7 +1,7 @@
 {
     "dataset": {
-        "repo_id": "aic-cable-insertion-tele-v174_no_zero_actions_state_filtered",
-        "root": "datasets/aic-cable-insertion-tele-v174_no_zero_actions_state_filtered",
+        "repo_id": "kejia98/aic-cable-insertion-tele-v180-state-processed_state_filtered",
+        "root": "datasets/aic-cable-insertion-tele-v180-state-processed_state_filtered",
         "episodes": null,
         "image_transforms": {
             "enable": false,
@@ -87,7 +87,7 @@
             "observation.state": {
                 "type": "STATE",
                 "shape": [
-                    7
+                    13
                 ]
             },
             "observation.images.left_camera": {
@@ -163,7 +163,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/ditflow_no_zero_actions_tele-v174_state",
+    "output_dir": "outputs/ditflow_tared_wrench_tele-v180_state_processed",
     "job_name": "ditflow",
     "resume": false,
     "seed": 1000,
@@ -204,7 +204,7 @@
         "project": "AICChallenge",
         "entity": null,
         "notes": null,
-        "run_id": "a6t4xru3",
+        "run_id": "wqukmai7",
         "mode": "online",
         "add_tags": true
     },