eperezs commited on
Commit
e642fd8
1 Parent(s): 5e0edff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -9,13 +9,14 @@ tags: []
9
 
10
 
11
 
12
- ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/66790f78c411cf3d84e7ceb0/owIYJZH3FK9VgUfloZ2Rc.jpeg)
 
13
  ## ---------
14
- checkpoint = "mental/mental-bert-base-uncased"
15
 
16
  learning_rate = 2e-5
17
 
18
- batch_size = 16
19
 
20
  num_train_epochs = 5
21
 
 
9
 
10
 
11
 
12
+
13
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/66790f78c411cf3d84e7ceb0/60cuWN1vE3e6UTBCt7S2p.png)
14
  ## ---------
15
+ checkpoint = "AIMH/mental-roberta-large"
16
 
17
  learning_rate = 2e-5
18
 
19
+ batch_size = 4
20
 
21
  num_train_epochs = 5
22