Update config.json
Browse files- config.json +3 -4
config.json
CHANGED
|
@@ -35,7 +35,7 @@
|
|
| 35 |
},
|
| 36 |
"device": "cuda",
|
| 37 |
"use_amp": false,
|
| 38 |
-
|
| 39 |
"push_to_hub": true,
|
| 40 |
"repo_id": "qownscks/pi0_widowx",
|
| 41 |
"private": null,
|
|
@@ -56,7 +56,7 @@
|
|
| 56 |
"time_sampling_offset": 0.001,
|
| 57 |
"min_period": 0.004,
|
| 58 |
"max_period": 4.0,
|
| 59 |
-
|
| 60 |
"image_resolution": [
|
| 61 |
224,
|
| 62 |
224
|
|
@@ -70,8 +70,7 @@
|
|
| 70 |
"gradient_checkpointing": true,
|
| 71 |
"compile_model": false,
|
| 72 |
"compile_mode": "max-autotune",
|
| 73 |
-
|
| 74 |
-
"train_expert_only": false,
|
| 75 |
"optimizer_lr": 2.5e-05,
|
| 76 |
"optimizer_betas": [
|
| 77 |
0.9,
|
|
|
|
| 35 |
},
|
| 36 |
"device": "cuda",
|
| 37 |
"use_amp": false,
|
| 38 |
+
|
| 39 |
"push_to_hub": true,
|
| 40 |
"repo_id": "qownscks/pi0_widowx",
|
| 41 |
"private": null,
|
|
|
|
| 56 |
"time_sampling_offset": 0.001,
|
| 57 |
"min_period": 0.004,
|
| 58 |
"max_period": 4.0,
|
| 59 |
+
|
| 60 |
"image_resolution": [
|
| 61 |
224,
|
| 62 |
224
|
|
|
|
| 70 |
"gradient_checkpointing": true,
|
| 71 |
"compile_model": false,
|
| 72 |
"compile_mode": "max-autotune",
|
| 73 |
+
|
|
|
|
| 74 |
"optimizer_lr": 2.5e-05,
|
| 75 |
"optimizer_betas": [
|
| 76 |
0.9,
|