Upload folder using huggingface_hub

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:af3dea5d665432ca98aedae9692cbc101499f5544e2ef4c206cba471f7c1c23c
+oid sha256:48d378be49ed9ad5b946c7e3c037a38e30d263474caff78cf94598270a522936
 size 206699736

train_config.json

@@ -77,7 +77,7 @@
         },
         "revision": null,
         "use_imagenet_stats": true,
-        "video_backend": "torchcodec",
+        "video_backend": "pyav",
         "return_uint8": false,
         "depth_output_unit": "mm",
         "streaming": false,
@@ -149,17 +149,17 @@
         "optimizer_lr_backbone": 1e-05
     },
     "reward_model": null,
-    "output_dir": "/content/drive/MyDrive/outputs/train/test_colab_v2",
-    "job_name": "test_training_airpod",
+    "output_dir": "/workspace/outputs/test_pick_place",
+    "job_name": "test_runpod",
     "resume": false,
     "seed": 1000,
     "cudnn_deterministic": false,
-    "num_workers": 2,
-    "batch_size": 8,
+    "num_workers": 4,
+    "batch_size": 16,
     "prefetch_factor": 4,
     "persistent_workers": true,
     "dataloader_multiprocessing_context": "spawn",
-    "steps": 10000,
+    "steps": 20000,
     "env_eval_freq": 20000,
     "log_freq": 100,
     "eval_steps": 0,
@@ -182,7 +182,7 @@
     "scheduler": null,
     "eval": {
         "n_episodes": 50,
-        "batch_size": 1,
+        "batch_size": 28,
         "use_async_envs": true,
         "recording": false,
         "recording_repo_id": null,
@@ -194,7 +194,7 @@
         "project": "lerobot-demo",
         "entity": null,
         "notes": null,
-        "run_id": "a6xqj271",
+        "run_id": "taho6uc7",
         "mode": null,
         "add_tags": true
     },