Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: diffusion
 pipeline_tag: robotics
 tags:
-- lerobot
 - robotics
 - diffusion
+- lerobot
 ---
 
 # Model Card for diffusion

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7d446b16968c4860305e397a2d99bb018e6e0de357d5b79cc68cc88c71744197
+oid sha256:45f7ff53af58575adf4dfe09d27c2035a0a5284ae9d2c1c79a4ffe55912fe16e
 size 312295392

train_config.json

@@ -175,14 +175,14 @@
     "resume": false,
     "seed": 1000,
     "cudnn_deterministic": false,
-    "num_workers": 24,
+    "num_workers": 12,
     "batch_size": 64,
-    "steps": 200,
+    "steps": 98505,
     "eval_freq": 20000,
     "log_freq": 3,
     "tolerance_s": 0.0001,
     "save_checkpoint": true,
-    "save_freq": 3284,
+    "save_freq": 8209,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adam",
@@ -211,7 +211,7 @@
         "project": "open-cloth",
         "entity": null,
         "notes": null,
-        "run_id": "do8q2lgi",
+        "run_id": "h1gajqhp",
         "mode": null,
         "add_tags": true
     },