Update config.json
Browse files- config.json +5 -5
config.json
CHANGED
|
@@ -48,8 +48,8 @@
|
|
| 48 |
"private": null,
|
| 49 |
"tags": null,
|
| 50 |
"license": null,
|
| 51 |
-
"chunk_size":
|
| 52 |
-
"n_action_steps":
|
| 53 |
"normalization_mapping": {
|
| 54 |
"VISUAL": "IDENTITY",
|
| 55 |
"STATE": "MEAN_STD",
|
|
@@ -70,14 +70,14 @@
|
|
| 70 |
"freeze_vision_encoder": true,
|
| 71 |
"train_expert_only": true,
|
| 72 |
"train_state_proj": true,
|
| 73 |
-
"optimizer_lr": 0.
|
| 74 |
"optimizer_betas": [
|
| 75 |
0.9,
|
| 76 |
0.95
|
| 77 |
],
|
| 78 |
"optimizer_eps": 1e-08,
|
| 79 |
-
"optimizer_weight_decay":
|
| 80 |
-
"optimizer_grad_clip_norm":
|
| 81 |
"scheduler_warmup_steps": 1000,
|
| 82 |
"scheduler_decay_steps": 30000,
|
| 83 |
"scheduler_decay_lr": 2.5e-06,
|
|
|
|
| 48 |
"private": null,
|
| 49 |
"tags": null,
|
| 50 |
"license": null,
|
| 51 |
+
"chunk_size": 30,
|
| 52 |
+
"n_action_steps": 20,
|
| 53 |
"normalization_mapping": {
|
| 54 |
"VISUAL": "IDENTITY",
|
| 55 |
"STATE": "MEAN_STD",
|
|
|
|
| 70 |
"freeze_vision_encoder": true,
|
| 71 |
"train_expert_only": true,
|
| 72 |
"train_state_proj": true,
|
| 73 |
+
"optimizer_lr": 0.00003,
|
| 74 |
"optimizer_betas": [
|
| 75 |
0.9,
|
| 76 |
0.95
|
| 77 |
],
|
| 78 |
"optimizer_eps": 1e-08,
|
| 79 |
+
"optimizer_weight_decay": 0.001,
|
| 80 |
+
"optimizer_grad_clip_norm": 5,
|
| 81 |
"scheduler_warmup_steps": 1000,
|
| 82 |
"scheduler_decay_steps": 30000,
|
| 83 |
"scheduler_decay_lr": 2.5e-06,
|