Upload folder using huggingface_hub

config.json

@@ -58,8 +58,8 @@
             ]
         }
     },
-    "chunk_size": 100,
-    "n_action_steps": 100,
+    "chunk_size": 50,
+    "n_action_steps": 25,
     "vision_backbone": "resnet18",
     "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
     "replace_final_stride_with_dilation": false,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0cb0554cf9d056162e3e2f49ebf7e0a77fe5cc548838b6b6baf04d248ea50e93
-size 206767840
+oid sha256:0f27280fdb8e5525207caace8fbb636465a607e1d245744e288a55c7723842e1
+size 206563040

train_config.json

@@ -124,8 +124,8 @@
                 ]
             }
         },
-        "chunk_size": 100,
-        "n_action_steps": 100,
+        "chunk_size": 50,
+        "n_action_steps": 25,
         "vision_backbone": "resnet18",
         "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
         "replace_final_stride_with_dilation": false,
@@ -146,19 +146,19 @@
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "outputs/train/2025-02-18/16-49-42_act",
+    "output_dir": "outputs/train/2025-02-19/19-32-39_act",
     "job_name": "act",
     "resume": false,
     "device": "cuda",
     "use_amp": false,
     "seed": 1000,
     "num_workers": 4,
-    "batch_size": 8,
-    "steps": 100000,
+    "batch_size": 16,
+    "steps": 1000000,
     "eval_freq": 20000,
     "log_freq": 200,
     "save_checkpoint": true,
-    "save_freq": 20000,
+    "save_freq": 5000,
     "use_policy_training_preset": true,
     "optimizer": {
         "type": "adamw",