Upload policy weights, train config and readme

README.md

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

config.json

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

model.safetensors

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

train_config.json

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