aframson commited on
Commit
7153f3d
·
1 Parent(s): fb451cc
Files changed (1) hide show
  1. 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
- "hidden_size": 300,
 
 
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