Upload policy weights, train config and readme

config.json

@@ -33,7 +33,7 @@
     "private": null,
     "tags": null,
     "license": null,
-    "pretrained_path": "/workspace/checkpoints/act-pusht-100k/checkpoints/100000/pretrained_model",
+    "pretrained_path": "/workspace/outputs/act-pusht-run3/checkpoints/last/pretrained_model",
     "chunk_size": 100,
     "n_action_steps": 100,
     "normalization_mapping": {

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e600e9435a448cfb0fcca6f3e8a25dd9f8634d8bb92c84ecda31f23d7abb5bf7
+oid sha256:cbb728006efef8d2e28f74dfde4c66173b1619069817babec9625b0dc7fa59f7
 size 206666936

train_config.json

@@ -4,7 +4,7 @@
         "root": null,
         "episodes": null,
         "image_transforms": {
-            "enable": false,
+            "enable": true,
             "max_num_transforms": 3,
             "random_order": false,
             "tfs": {
@@ -115,7 +115,7 @@
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": "/workspace/checkpoints/act-pusht-100k/checkpoints/100000/pretrained_model",
+        "pretrained_path": "/workspace/outputs/act-pusht-run3/checkpoints/last/pretrained_model",
         "chunk_size": 100,
         "n_action_steps": 100,
         "normalization_mapping": {
@@ -143,13 +143,13 @@
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "/workspace/outputs/act-pusht-run3",
+    "output_dir": "/workspace/outputs/act-pusht-run4",
     "job_name": "act_pusht_job",
     "resume": false,
     "seed": 1000,
     "num_workers": 4,
     "batch_size": 8,
-    "steps": 200000,
+    "steps": 400000,
     "eval_freq": 10000,
     "log_freq": 200,
     "tolerance_s": 0.0001,
@@ -179,7 +179,7 @@
         "project": "pusht-t4",
         "entity": null,
         "notes": null,
-        "run_id": "4fsg80hh",
+        "run_id": "ckiaz6no",
         "mode": null
     },
     "peft": null,