Upload policy weights, train config and readme

README.md

@@ -1,13 +1,13 @@
 ---
-datasets: Task_1_2_dataset_delta_pose_09051205_merged_approach60_insert_only_tcp_state
+datasets: mirmiAIC/Task_1_2_3_all_merged_oliver_manually_processed
 library_name: lerobot
 license: apache-2.0
 model_name: ditflow
 pipeline_tag: robotics
 tags:
-- ditflow
-- robotics
 - lerobot
+- robotics
+- ditflow
 ---
 
 # Model Card for ditflow

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a3f91f3582288ec150da5dd50dec5a8641197996f0fe7bcdb91e2f92dc0e0f67
+oid sha256:62103fce972c526e467d396dc8714d3b0424abd8497438fbea4324a38d77364b
 size 257806640

train_config.json

@@ -1,7 +1,7 @@
 {
     "dataset": {
-        "repo_id": "Task_1_2_dataset_delta_pose_09051205_merged_approach60_insert_only_tcp_state",
-        "root": "datasets/mirmiAIC/Task_1_2_dataset_delta_pose_09051205_merged_approach60_insert_only_tcp_state",
+        "repo_id": "mirmiAIC/Task_1_2_3_all_merged_oliver_manually_processed",
+        "root": "datasets/mirmiAIC/mirmiAIC/Task_1_2_3_all_merged_oliver_manually_processed",
         "episodes": null,
         "image_transforms": {
             "enable": false,
@@ -166,7 +166,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/ditflow_no_cart_rel_act",
+    "output_dir": "outputs/ditflow_all",
     "job_name": "ditflow",
     "resume": false,
     "seed": 1000,
@@ -178,7 +178,7 @@
     "log_freq": 2000,
     "tolerance_s": 0.0001,
     "save_checkpoint": true,
-    "save_freq": 25000,
+    "save_freq": 10000,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adam",
@@ -207,7 +207,7 @@
         "project": "AICChallenge",
         "entity": null,
         "notes": null,
-        "run_id": "qhpfym81",
+        "run_id": "toby626e",
         "mode": "online",
         "add_tags": true
     },