Upload policy weights, train config and readme

README.md

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

config.json

@@ -33,7 +33,7 @@
             ]
         }
     },
-    "device": "mps",
+    "device": "cuda",
     "use_amp": false,
     "push_to_hub": true,
     "repo_id": "yuweiiizz/il_gym_act_policy",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:da0fc5c086c72c50745467d6356f2ba38f64dbd88da460a9d1812a2f31850aa3
+oid sha256:2fe1397ecea44c8b3a30fea65b917810cbdb30854a29faaab47ee2722e7c2911
 size 206740680

train_config.json

@@ -76,7 +76,7 @@
         },
         "revision": null,
         "use_imagenet_stats": true,
-        "video_backend": "pyav",
+        "video_backend": "torchcodec",
         "streaming": false
     },
     "env": null,
@@ -115,7 +115,7 @@
                 ]
             }
         },
-        "device": "mps",
+        "device": "cuda",
         "use_amp": false,
         "push_to_hub": true,
         "repo_id": "yuweiiizz/il_gym_act_policy",
@@ -150,7 +150,7 @@
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "outputs/train/il_sim_test",
+    "output_dir": "outputs/train/il_sim_test0",
     "job_name": "il_sim_test",
     "resume": false,
     "seed": 1000,
@@ -185,7 +185,7 @@
         "project": "lerobot",
         "entity": null,
         "notes": null,
-        "run_id": "hsgj5ix3",
+        "run_id": "6oo7ppwr",
         "mode": null
     },
     "checkpoint_path": null,