Upload folder using huggingface_hub

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1a039cd06220642dd5812b17037afd0870515f61a47794c02ea9b301977a5659
+oid sha256:0b1d2ccb891a9b5d35294797563d8ffb5941575256bdc949be517bdca40da75b
 size 906712520

policy_postprocessor_step_0_unnormalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c5f65cdb47897e7a030860d390f50d38b2a8a3a39a7b971302bae18a725ab9c8
+oid sha256:6cc9821e2ba63573f7e3f0fac380d6376dcb62ff8fbd5649437db67fdcf5c978
 size 4000

policy_preprocessor_step_5_normalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c5f65cdb47897e7a030860d390f50d38b2a8a3a39a7b971302bae18a725ab9c8
+oid sha256:6cc9821e2ba63573f7e3f0fac380d6376dcb62ff8fbd5649437db67fdcf5c978
 size 4000

train_config.json

@@ -74,7 +74,7 @@
                 }
             }
         },
-        "revision": null,
+        "revision": "main",
         "use_imagenet_stats": true,
         "video_backend": "torchcodec",
         "streaming": false
@@ -183,8 +183,8 @@
     "resume": false,
     "seed": 1000,
     "num_workers": 4,
-    "batch_size": 16,
-    "steps": 20000,
+    "batch_size": 64,
+    "steps": 50000,
     "eval_freq": 20000,
     "log_freq": 200,
     "save_checkpoint": true,