Upload policy weights, train config and readme

README.md

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3e708f20b1f4cda117a1263b3d108dbc4ff125dc7ddc78fca8d5347aa5657f09
+oid sha256:1031655cd59dfb683ef99f73a4224753419f505104f2f92d5ff8f814f9b5eeb8
 size 1053834112

train_config.json

@@ -214,12 +214,12 @@
     "batch_size": 8,
     "prefetch_factor": 4,
     "persistent_workers": true,
-    "steps": 5,
-    "eval_freq": 2,
+    "steps": 1000,
+    "eval_freq": 500,
     "log_freq": 200,
     "tolerance_s": 0.0001,
     "save_checkpoint": true,
-    "save_freq": 1,
+    "save_freq": 500,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adam",
@@ -245,10 +245,10 @@
     "wandb": {
         "enable": true,
         "disable_artifact": false,
-        "project": "lerobot",
+        "project": "UADP-baselines-tmp",
         "entity": null,
         "notes": null,
-        "run_id": "gtb58yra",
+        "run_id": "diffusion_aloha_transfer_cube_seed42",
         "mode": null,
         "add_tags": true
     },