cxc
Browse files- config.json +3 -3
config.json
CHANGED
@@ -11,11 +11,11 @@
|
|
11 |
"device": "cpu",
|
12 |
"eval_interval": 250,
|
13 |
"hidden_dropout_prob": 0.0,
|
14 |
-
"
|
|
|
|
|
15 |
"learning_rate": 0.001,
|
16 |
"max_iters": 6000,
|
17 |
-
"num_attention_heads": 6,
|
18 |
-
"num_hidden_layers": 6,
|
19 |
"torch_dtype": "float16",
|
20 |
"transformers_version": "4.33.2",
|
21 |
"vocab_size": 1000
|
|
|
11 |
"device": "cpu",
|
12 |
"eval_interval": 250,
|
13 |
"hidden_dropout_prob": 0.0,
|
14 |
+
"n_embd": 300,
|
15 |
+
"n_head": 6,
|
16 |
+
"n_layer": 6,
|
17 |
"learning_rate": 0.001,
|
18 |
"max_iters": 6000,
|
|
|
|
|
19 |
"torch_dtype": "float16",
|
20 |
"transformers_version": "4.33.2",
|
21 |
"vocab_size": 1000
|