Upload 3 files

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3df7f14d0545dfaecf7ab4ac2998164debcf271259a647deee3c57fbc853a5e7
+oid sha256:8d38224742fabd7f1843a4dc24eb050c19a317fccef1325ccc5a8ee4fd87722a
 size 906713296

train_config.json

@@ -1,7 +1,7 @@
 {
     "dataset": {
         "repo_id": "MulixBF/record-cube-pick-2cam-beige-2",
-        "root": "MulixBF/",
+        "root": null,
         "episodes": null,
         "image_transforms": {
             "enable": false,
@@ -148,13 +148,13 @@
         "min_period": 0.004,
         "max_period": 4.0
     },
-    "output_dir": "outputs/train/colored_cubes",
+    "output_dir": "outputs/train/colored_cubes_v2",
     "job_name": "colored_cubes_training",
     "resume": false,
     "seed": 1000,
     "num_workers": 4,
-    "batch_size": 5,
-    "steps": 2000,
+    "batch_size": 8,
+    "steps": 4000,
     "eval_freq": 20000,
     "log_freq": 200,
     "save_checkpoint": true,