Upload folder using huggingface_hub

config.json

@@ -31,9 +31,9 @@
         }
     },
     "device": "cuda",
-    "use_amp": false,
+    "use_amp": true,
     "push_to_hub": true,
-    "repo_id": "U-RIL/my-policy",
+    "repo_id": "U-RIL/my_policy",
     "private": null,
     "tags": null,
     "license": null,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fa2e174d051a0629b110e5eb046bcea0615d3fc794e02d9af7da5c3874ec18b2
+oid sha256:6a4796d84088c10c521729ffd9be7ac1fb19f512fc1e793cc1b04c3e1e9493b3
 size 206700792

train_config.json

@@ -1,6 +1,6 @@
 {
     "dataset": {
-        "repo_id": "U-RIL/record-test",
+        "repo_id": "U-RIL/record-Pastery",
         "root": null,
         "episodes": null,
         "image_transforms": {
@@ -98,9 +98,9 @@
             }
         },
         "device": "cuda",
-        "use_amp": false,
+        "use_amp": true,
         "push_to_hub": true,
-        "repo_id": "U-RIL/my-policy",
+        "repo_id": "U-RIL/my_policy",
         "private": null,
         "tags": null,
         "license": null,
@@ -126,17 +126,17 @@
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "outputs/train/il_sim_test0",
-    "job_name": "il_sim_test",
+    "output_dir": "outputs/train/il_sim_test4",
+    "job_name": "train-Pastery",
     "resume": false,
     "seed": 1000,
     "num_workers": 4,
     "batch_size": 16,
-    "steps": 15000,
-    "eval_freq": 3000,
+    "steps": 22000,
+    "eval_freq": 2000,
     "log_freq": 200,
     "save_checkpoint": true,
-    "save_freq": 2000,
+    "save_freq": 20000,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adamw",
@@ -156,12 +156,12 @@
         "use_async_envs": false
     },
     "wandb": {
-        "enable": false,
+        "enable": true,
         "disable_artifact": false,
         "project": "lerobot",
         "entity": null,
         "notes": null,
-        "run_id": null,
+        "run_id": "np0tvvky",
         "mode": null
     }
 }