underwater45 commited on
Commit
c864715
·
verified ·
1 Parent(s): ebd9bf3

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +32 -0
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Magical-1.5_star",
3
+ "model_type": "transformer",
4
+ "architecture": {
5
+ "vocab_size": 30522,
6
+ "d_model": 512,
7
+ "nhead": 8,
8
+ "num_layers": 4,
9
+ "dim_feedforward": 2048,
10
+ "max_len": 128,
11
+ "dropout": 0.1
12
+ },
13
+ "training": {
14
+ "batch_size": 2,
15
+ "num_epochs": 3,
16
+ "learning_rate": 5e-5,
17
+ "optimizer": "AdamW",
18
+ "loss_function": "CrossEntropyLoss"
19
+ },
20
+ "data": {
21
+ "tokenizer": "bert-base-uncased",
22
+ "max_length": 128,
23
+ "dataset": {
24
+ "name": "SimpleChatDataset"
25
+ ]
26
+ }
27
+ },
28
+ "save": {
29
+ "format": "safetensors",
30
+ "path": "magical_1.5_star.safetensors"
31
+ }
32
+ }