Upload folder using huggingface_hub

config.json

@@ -44,7 +44,7 @@
     "device": "cuda",
     "use_amp": true,
     "push_to_hub": true,
-    "repo_id": "Gokulavelan/Water-Pump-Assembly",
+    "repo_id": "Gokulavelan/Water-Pump-Assembly-V2",
     "private": null,
     "tags": null,
     "license": null,

model.safetensors

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

policy_postprocessor_step_0_unnormalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d814322a1a480c369606d1a60810db34aaf171981b73bde9f4f1f26144229ecf
+oid sha256:5169b83b77ec30edf3ede5e34de13301d74bc2c017f36d02ba239c6d3e16c32a
 size 7552

policy_preprocessor_step_5_normalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d814322a1a480c369606d1a60810db34aaf171981b73bde9f4f1f26144229ecf
+oid sha256:5169b83b77ec30edf3ede5e34de13301d74bc2c017f36d02ba239c6d3e16c32a
 size 7552

train_config.json

@@ -1,7 +1,7 @@
 {
     "dataset": {
-        "repo_id": "Gokulavelan/Water-Pump-Assembly",
-        "root": "/home/dewa/.cache/huggingface/hub/datasets--Gokulavelan--Water-Pump-Assembly/snapshots/992b74289d74e0e64df58219c39103488be7ddca",
+        "repo_id": "Gokulavelan/Water-Pump-Assembly-V2",
+        "root": "/home/dewa/.cache/huggingface/hub/datasets--Gokulavelan--Water-Pump-Assembly-v2/snapshots/2699e273590ff658b7305da8e354622b1a2f9cfd",
         "episodes": null,
         "image_transforms": {
             "enable": false,
@@ -126,7 +126,7 @@
         "device": "cuda",
         "use_amp": true,
         "push_to_hub": true,
-        "repo_id": "Gokulavelan/Water-Pump-Assembly",
+        "repo_id": "Gokulavelan/Water-Pump-Assembly-V2",
         "private": null,
         "tags": null,
         "license": null,
@@ -177,13 +177,13 @@
         "min_period": 0.004,
         "max_period": 4.0
     },
-    "output_dir": "outputs/train/water_pump_vla",
+    "output_dir": "outputs/train/water_pump_vla_0512",
     "job_name": "my_smolvla_training_accel",
     "resume": false,
     "seed": 1000,
     "num_workers": 2,
-    "batch_size": 32,
-    "steps": 12000,
+    "batch_size": 64,
+    "steps": 30000,
     "eval_freq": 20000,
     "log_freq": 200,
     "save_checkpoint": true,