Update gsv/config.json
Browse files- gsv/config.json +2 -25
gsv/config.json
CHANGED
|
@@ -1,24 +1,6 @@
|
|
| 1 |
{
|
| 2 |
-
"train": {
|
| 3 |
-
"seed": 1234,
|
| 4 |
-
"epochs": 200,
|
| 5 |
-
"batch_size": 5,
|
| 6 |
-
"save_every_n_epoch": 1,
|
| 7 |
-
"precision": "16-mixed",
|
| 8 |
-
"gradient_clip": 1.0
|
| 9 |
-
},
|
| 10 |
-
"optimizer": {
|
| 11 |
-
"lr": 0.01,
|
| 12 |
-
"lr_init": 1e-05,
|
| 13 |
-
"lr_end": 0.0001,
|
| 14 |
-
"warmup_steps": 2000,
|
| 15 |
-
"decay_steps": 40000
|
| 16 |
-
},
|
| 17 |
"data": {
|
| 18 |
-
"
|
| 19 |
-
"max_sec": 57,
|
| 20 |
-
"num_workers": 4,
|
| 21 |
-
"pad_val": 1024
|
| 22 |
},
|
| 23 |
"model": {
|
| 24 |
"vocab_size": 1025,
|
|
@@ -26,13 +8,8 @@
|
|
| 26 |
"embedding_dim": 512,
|
| 27 |
"hidden_dim": 512,
|
| 28 |
"head": 16,
|
| 29 |
-
"linear_units": 2048,
|
| 30 |
"n_layer": 24,
|
| 31 |
"dropout": 0,
|
| 32 |
-
"EOS": 1024
|
| 33 |
-
"random_bert": 0
|
| 34 |
-
},
|
| 35 |
-
"inference": {
|
| 36 |
-
"top_k": 5
|
| 37 |
}
|
| 38 |
}
|
|
|
|
| 1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
"data": {
|
| 3 |
+
"max_sec": 57
|
|
|
|
|
|
|
|
|
|
| 4 |
},
|
| 5 |
"model": {
|
| 6 |
"vocab_size": 1025,
|
|
|
|
| 8 |
"embedding_dim": 512,
|
| 9 |
"hidden_dim": 512,
|
| 10 |
"head": 16,
|
|
|
|
| 11 |
"n_layer": 24,
|
| 12 |
"dropout": 0,
|
| 13 |
+
"EOS": 1024
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
}
|
| 15 |
}
|