Upload policy weights, train config and readme

README.md

@@ -6,8 +6,8 @@ model_name: diffusion
 pipeline_tag: robotics
 tags:
 - lerobot
-- robotics
 - diffusion
+- robotics
 ---
 
 # Model Card for diffusion

config.json

@@ -44,7 +44,7 @@
     "device": "cuda",
     "use_amp": false,
     "push_to_hub": true,
-    "repo_id": "ISdept/piper-act",
+    "repo_id": "ISdept/piper_arm",
     "private": null,
     "tags": null,
     "license": null,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:02574ea7a953bdf2042461b7f1227459b53c72740539666cbd63a981d95e6dd0
+oid sha256:5f285fb264582e1e93c518db1bb296189b26e5e8282a84007432408fb98a8e56
 size 1081440164

train_config.json

@@ -126,7 +126,7 @@
         "device": "cuda",
         "use_amp": false,
         "push_to_hub": true,
-        "repo_id": "ISdept/piper-act",
+        "repo_id": "ISdept/piper_arm",
         "private": null,
         "tags": null,
         "license": null,
@@ -178,7 +178,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/output_act",
+    "output_dir": "outputs/train/output_diffusion",
     "job_name": "pick_and_place",
     "resume": false,
     "seed": 1000,