Upload policy weights, train config and readme
Browse files- config.json +2 -1
- model.safetensors +1 -1
- train_config.json +3 -3
config.json
CHANGED
|
@@ -27,6 +27,7 @@
|
|
| 27 |
},
|
| 28 |
"device": "cuda",
|
| 29 |
"use_amp": false,
|
|
|
|
| 30 |
"push_to_hub": true,
|
| 31 |
"repo_id": "SagriviS/Test",
|
| 32 |
"private": null,
|
|
@@ -55,7 +56,7 @@
|
|
| 55 |
"n_vae_encoder_layers": 4,
|
| 56 |
"temporal_ensemble_coeff": null,
|
| 57 |
"dropout": 0.1,
|
| 58 |
-
"kl_weight":
|
| 59 |
"optimizer_lr": 1e-05,
|
| 60 |
"optimizer_weight_decay": 0.0001,
|
| 61 |
"optimizer_lr_backbone": 1e-05
|
|
|
|
| 27 |
},
|
| 28 |
"device": "cuda",
|
| 29 |
"use_amp": false,
|
| 30 |
+
"use_peft": false,
|
| 31 |
"push_to_hub": true,
|
| 32 |
"repo_id": "SagriviS/Test",
|
| 33 |
"private": null,
|
|
|
|
| 56 |
"n_vae_encoder_layers": 4,
|
| 57 |
"temporal_ensemble_coeff": null,
|
| 58 |
"dropout": 0.1,
|
| 59 |
+
"kl_weight": 1.0,
|
| 60 |
"optimizer_lr": 1e-05,
|
| 61 |
"optimizer_weight_decay": 0.0001,
|
| 62 |
"optimizer_lr_backbone": 1e-05
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 206970104
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:43d609279b37579e8dc324509863621d739d69d9aacc3111750fbda6a73185be
|
| 3 |
size 206970104
|
train_config.json
CHANGED
|
@@ -138,12 +138,12 @@
|
|
| 138 |
"n_vae_encoder_layers": 4,
|
| 139 |
"temporal_ensemble_coeff": null,
|
| 140 |
"dropout": 0.1,
|
| 141 |
-
"kl_weight":
|
| 142 |
"optimizer_lr": 1e-05,
|
| 143 |
"optimizer_weight_decay": 0.0001,
|
| 144 |
"optimizer_lr_backbone": 1e-05
|
| 145 |
},
|
| 146 |
-
"output_dir": "lerobot/outputs/train/
|
| 147 |
"job_name": "act",
|
| 148 |
"resume": false,
|
| 149 |
"seed": 1000,
|
|
@@ -179,7 +179,7 @@
|
|
| 179 |
"project": "lerobot",
|
| 180 |
"entity": null,
|
| 181 |
"notes": null,
|
| 182 |
-
"run_id": "
|
| 183 |
"mode": null
|
| 184 |
},
|
| 185 |
"peft": null,
|
|
|
|
| 138 |
"n_vae_encoder_layers": 4,
|
| 139 |
"temporal_ensemble_coeff": null,
|
| 140 |
"dropout": 0.1,
|
| 141 |
+
"kl_weight": 1.0,
|
| 142 |
"optimizer_lr": 1e-05,
|
| 143 |
"optimizer_weight_decay": 0.0001,
|
| 144 |
"optimizer_lr_backbone": 1e-05
|
| 145 |
},
|
| 146 |
+
"output_dir": "lerobot/outputs/train/Test14",
|
| 147 |
"job_name": "act",
|
| 148 |
"resume": false,
|
| 149 |
"seed": 1000,
|
|
|
|
| 179 |
"project": "lerobot",
|
| 180 |
"entity": null,
|
| 181 |
"notes": null,
|
| 182 |
+
"run_id": "rkh5wl7v",
|
| 183 |
"mode": null
|
| 184 |
},
|
| 185 |
"peft": null,
|