Upload folder using huggingface_hub

config.json

@@ -8,7 +8,7 @@
                 6
             ]
         },
-        "observation.images.front": {
+        "observation.images.wrist": {
             "type": "VISUAL",
             "shape": [
                 3,
@@ -16,7 +16,7 @@
                 640
             ]
         },
-        "observation.images.wrist": {
+        "observation.images.front": {
             "type": "VISUAL",
             "shape": [
                 3,
@@ -36,14 +36,14 @@
     "device": "cuda",
     "use_amp": false,
     "use_peft": false,
-    "push_to_hub": false,
-    "repo_id": null,
+    "push_to_hub": true,
+    "repo_id": "Oiki123/act_pick_noaug_50k",
     "private": null,
     "tags": null,
     "license": null,
     "pretrained_path": null,
-    "chunk_size": 100,
-    "n_action_steps": 100,
+    "chunk_size": 50,
+    "n_action_steps": 15,
     "normalization_mapping": {
         "VISUAL": "MEAN_STD",
         "STATE": "MEAN_STD",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b0f096bafd9126c8ecd6a93d3b93beff0acda6d375c73a8f2cabe5ddc08336b4
-size 206699736
+oid sha256:f8fecaf62d928a29bf10c25c81403f7813e874d5d476b9a06fe2872dcc93ed83
+size 206494928

policy_postprocessor_step_0_unnormalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e005991c93724b721438f9d1c6b317d3e083c2827befc531dc41786ab6dfa2d
+oid sha256:0a9330844c8106105b3aba7c98c5828004e6f9678c0be2ff45235e2da4b66bf5
 size 7552

policy_preprocessor.json

@@ -29,7 +29,7 @@
               6
             ]
           },
-          "observation.images.front": {
+          "observation.images.wrist": {
             "type": "VISUAL",
             "shape": [
               3,
@@ -37,7 +37,7 @@
               640
             ]
           },
-          "observation.images.wrist": {
+          "observation.images.front": {
             "type": "VISUAL",
             "shape": [
               3,

policy_preprocessor_step_3_normalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e005991c93724b721438f9d1c6b317d3e083c2827befc531dc41786ab6dfa2d
+oid sha256:0a9330844c8106105b3aba7c98c5828004e6f9678c0be2ff45235e2da4b66bf5
 size 7552

train_config.json

@@ -1,6 +1,6 @@
 {
     "dataset": {
-        "repo_id": "Shrek0/pick_10_pt2",
+        "repo_id": "Oiki123/so101_pick_merged",
         "root": null,
         "episodes": null,
         "image_transforms": {
@@ -90,7 +90,7 @@
                     6
                 ]
             },
-            "observation.images.front": {
+            "observation.images.wrist": {
                 "type": "VISUAL",
                 "shape": [
                     3,
@@ -98,7 +98,7 @@
                     640
                 ]
             },
-            "observation.images.wrist": {
+            "observation.images.front": {
                 "type": "VISUAL",
                 "shape": [
                     3,
@@ -118,14 +118,14 @@
         "device": "cuda",
         "use_amp": false,
         "use_peft": false,
-        "push_to_hub": false,
-        "repo_id": null,
+        "push_to_hub": true,
+        "repo_id": "Oiki123/act_pick_noaug_50k",
         "private": null,
         "tags": null,
         "license": null,
         "pretrained_path": null,
-        "chunk_size": 100,
-        "n_action_steps": 100,
+        "chunk_size": 50,
+        "n_action_steps": 15,
         "normalization_mapping": {
             "VISUAL": "MEAN_STD",
             "STATE": "MEAN_STD",
@@ -151,12 +151,12 @@
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "/root/outputs/act_no_aug",
-    "job_name": "act",
+    "output_dir": "outputs/act_pick_noaug",
+    "job_name": "act_pick_noaug",
     "resume": false,
     "seed": 1000,
-    "num_workers": 16,
-    "batch_size": 32,
+    "num_workers": 4,
+    "batch_size": 8,
     "steps": 50000,
     "eval_freq": 20000,
     "log_freq": 100,
@@ -183,8 +183,8 @@
     },
     "wandb": {
         "enable": false,
-        "disable_artifact": true,
-        "project": "lerobot",
+        "disable_artifact": false,
+        "project": "hackathon",
         "entity": null,
         "notes": null,
         "run_id": null,