Upload config.json
Browse files- config.json +3 -3
config.json
CHANGED
|
@@ -128,10 +128,10 @@
|
|
| 128 |
"PATIENCE": 3,
|
| 129 |
"DELTA": 0.001,
|
| 130 |
"VERBOSE": true,
|
| 131 |
-
"NUM_EPOCHS":
|
| 132 |
"LEARNING_RATE": 0.0001,
|
| 133 |
"TRANSFORMER_LEARNING_RATE": 5e-5,
|
| 134 |
-
"TRANSFORMER_NUM_EPOCHS":
|
| 135 |
},
|
| 136 |
"CHECKPOINT_CONFIG": {
|
| 137 |
"PATH": "checkpoints",
|
|
@@ -192,7 +192,7 @@
|
|
| 192 |
"TRANSFORMER_FILENAME_FORMAT": "transformer_model_{specialization}_{epoch}.pt",
|
| 193 |
"MODEL_NAME": "gpt2",
|
| 194 |
"MAX_SEQ_LENGTH": 767,
|
| 195 |
-
"NUM_EPOCHS":
|
| 196 |
"LEARNING_RATE": 5e-5,
|
| 197 |
"BATCH_SIZE": 32,
|
| 198 |
"EMBEDDING_DIM": 768,
|
|
|
|
| 128 |
"PATIENCE": 3,
|
| 129 |
"DELTA": 0.001,
|
| 130 |
"VERBOSE": true,
|
| 131 |
+
"NUM_EPOCHS": 20,
|
| 132 |
"LEARNING_RATE": 0.0001,
|
| 133 |
"TRANSFORMER_LEARNING_RATE": 5e-5,
|
| 134 |
+
"TRANSFORMER_NUM_EPOCHS": 20
|
| 135 |
},
|
| 136 |
"CHECKPOINT_CONFIG": {
|
| 137 |
"PATH": "checkpoints",
|
|
|
|
| 192 |
"TRANSFORMER_FILENAME_FORMAT": "transformer_model_{specialization}_{epoch}.pt",
|
| 193 |
"MODEL_NAME": "gpt2",
|
| 194 |
"MAX_SEQ_LENGTH": 767,
|
| 195 |
+
"NUM_EPOCHS": 20,
|
| 196 |
"LEARNING_RATE": 5e-5,
|
| 197 |
"BATCH_SIZE": 32,
|
| 198 |
"EMBEDDING_DIM": 768,
|