underwater45 commited on
Commit
01c9751
·
verified ·
1 Parent(s): 64ae298

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -0
config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Magical-1 Sun",
3
+ "model_type": "bert",
4
+ "base_model": "bert-base-uncased",
5
+ "num_labels": 2,
6
+ "max_length": 128,
7
+ "batch_size": 2,
8
+ "learning_rate": 2e-5,
9
+ "epochs": 3,
10
+ "device": "cuda"
11
+ }