WildnerveAI commited on
Commit
928a969
·
verified ·
1 Parent(s): 26929f2

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -128,10 +128,10 @@
128
  "PATIENCE": 3,
129
  "DELTA": 0.001,
130
  "VERBOSE": true,
131
- "NUM_EPOCHS": 10,
132
  "LEARNING_RATE": 0.0001,
133
  "TRANSFORMER_LEARNING_RATE": 5e-5,
134
- "TRANSFORMER_NUM_EPOCHS": 5
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": 10,
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,