Upload train_config.json (v0.4.0 ckpt-18k)

train_config.json

@@ -1,7 +1,7 @@
 {
     "dataset": {
         "repo_id": "LightwheelAI/leisaac-pick-orange",
-        "root": "/home/david/work/LeIsaac/datasets/raw/leisaac-pick-orange",
+        "root": "/home/david/work/isaaclab-experience/LeIsaac/datasets/raw/leisaac-pick-orange",
         "episodes": null,
         "image_transforms": {
             "enable": false,
@@ -77,7 +77,6 @@
         "revision": null,
         "use_imagenet_stats": true,
         "video_backend": "pyav",
-        "return_uint8": false,
         "streaming": false
     },
     "env": null,
@@ -118,13 +117,12 @@
         },
         "device": "cuda",
         "use_amp": false,
-        "use_peft": false,
         "push_to_hub": false,
         "repo_id": null,
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": null,
+        "pretrained_path": "LeIsaac/outputs/act-v040-baseline/checkpoints/last/pretrained_model",
         "chunk_size": 100,
         "n_action_steps": 100,
         "normalization_mapping": {
@@ -134,7 +132,7 @@
         },
         "vision_backbone": "resnet18",
         "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
-        "replace_final_stride_with_dilation": false,
+        "replace_final_stride_with_dilation": 0,
         "pre_norm": false,
         "dim_model": 512,
         "n_heads": 8,
@@ -152,19 +150,15 @@
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "/home/david/work/LeIsaac/outputs/act-leisaac-pick-orange",
-    "job_name": "act-leisaac-pick-orange",
-    "resume": false,
+    "output_dir": "LeIsaac/outputs/act-v040-baseline",
+    "job_name": "act-v040-baseline",
+    "resume": true,
     "seed": 1000,
-    "cudnn_deterministic": false,
     "num_workers": 4,
     "batch_size": 8,
-    "prefetch_factor": 4,
-    "persistent_workers": true,
-    "steps": 10000,
+    "steps": 20000,
     "eval_freq": 20000,
     "log_freq": 200,
-    "tolerance_s": 0.0001,
     "save_checkpoint": true,
     "save_freq": 2000,
     "use_policy_training_preset": true,
@@ -182,8 +176,8 @@
     "scheduler": null,
     "eval": {
         "n_episodes": 50,
-        "batch_size": 22,
-        "use_async_envs": true
+        "batch_size": 50,
+        "use_async_envs": false
     },
     "wandb": {
         "enable": false,
@@ -192,15 +186,8 @@
         "entity": null,
         "notes": null,
         "run_id": null,
-        "mode": null,
-        "add_tags": true
+        "mode": null
     },
-    "peft": null,
-    "use_rabc": false,
-    "rabc_progress_path": null,
-    "rabc_kappa": 0.01,
-    "rabc_epsilon": 1e-06,
-    "rabc_head_mode": "sparse",
-    "rename_map": {},
-    "checkpoint_path": null
+    "checkpoint_path": "LeIsaac/outputs/act-v040-baseline/checkpoints/last",
+    "rename_map": {}
 }