Upload train_args.json

storage/sigma_lora_out/train_args.json

@@ -2,11 +2,11 @@
   "sigma_env": "sigma.env",
   "output_dir": "/workspace/storage/sigma_lora_out",
   "base_model_id": "lerobot/pi05_base",
-  "load_in_4bit": false,
+  "load_in_4bit": true,
   "torch_dtype": "bf16",
-  "dataset_id": "lerobot/svla_so101_pickplace",
+  "dataset_id": null,
   "split": "train",
-  "data_dir": "/workspace/storage/sigma_pickplace",
+  "data_dir": "/workspace/.hf_sigma_cache/Veltraxor__Sigma/storage/sigma_pickplace",
   "hf_data_repo": "Veltraxor/Sigma",
   "hf_data_subdir": "storage/sigma_pickplace",
   "prefer_hf_shards": false,
@@ -14,7 +14,7 @@
   "hf_cache_keep_latest": 1,
   "hf_cache_dir": null,
   "num_workers": 4,
-  "epochs": 1,
+  "epochs": 3,
   "batch_size": 2,
   "grad_accum": 8,
   "lr": 0.0002,
@@ -28,7 +28,7 @@
   "alpha_c": 1.0,
   "lambda_sem": 1.2,
   "lambda_intent": 0.8,
-  "lambda_tau": 0.03,
+  "lambda_tau": 0.05,
   "beta_mi": 0.1,
   "eta_var": 0.1,
   "hard_mining_ratio": 0.3,