Update README.md
Browse files
README.md
CHANGED
@@ -9,13 +9,14 @@ tags: []
|
|
9 |
|
10 |
|
11 |
|
12 |
-
|
|
|
13 |
## ---------
|
14 |
-
checkpoint = "
|
15 |
|
16 |
learning_rate = 2e-5
|
17 |
|
18 |
-
batch_size =
|
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 |
|