Upload policy weights, train config and readme

config.json

@@ -76,7 +76,8 @@
         "weights": "./vjepa2_vitl.pt",
         "patch_size": 16,
         "temporal_offset": 30,
-        "use_learned_pos_embed": false
+        "use_learned_pos_embed": false,
+        "use_patch_merge": true
     },
     "sam3": null
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7a1963c640482aac38240b678c869cbdd6a80586779ecf9e757d2e43f58b6614
-size 1379231516
+oid sha256:8cd7d461e6c06f1b7b0f74ba477d6ef77decaedd4219d6ecc53c40eb4c5ee3f7
+size 1383428092

train_config.json

@@ -2,7 +2,28 @@
     "dataset": {
         "repo_id": "danaaubakirova/so100_task_2",
         "root": null,
-        "episodes": null,
+        "episodes": [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15,
+            16,
+            17,
+            18,
+            19
+        ],
         "image_transforms": {
             "enable": false,
             "max_num_transforms": 3,
@@ -143,7 +164,8 @@
             "weights": "./vjepa2_vitl.pt",
             "patch_size": 16,
             "temporal_offset": 30,
-            "use_learned_pos_embed": false
+            "use_learned_pos_embed": false,
+            "use_patch_merge": true
         },
         "sam3": null
     },
@@ -152,7 +174,7 @@
     "resume": false,
     "seed": 1000,
     "num_workers": 4,
-    "batch_size": 1,
+    "batch_size": 8,
     "steps": 1000,
     "eval_freq": 20000,
     "log_freq": 200,