Update README.md
Browse files
README.md
CHANGED
@@ -22,7 +22,7 @@ It has been trained to recognize six classes of named entities:
|
|
22 |
- FIRM: name of firm affiliated with inventor
|
23 |
- COMM: name and information mentioned about communicant
|
24 |
|
25 |
-
We take the original xlm-roberta-large [weights](https://huggingface.co/FacebookAI/xlm-roberta-large/blob/main/pytorch_model.bin) and fine tune on our custom dataset for 29 epochs with a learning rate of 5e-05 and a batch size of
|
26 |
|
27 |
### Usage
|
28 |
|
|
|
22 |
- FIRM: name of firm affiliated with inventor
|
23 |
- COMM: name and information mentioned about communicant
|
24 |
|
25 |
+
We take the original xlm-roberta-large [weights](https://huggingface.co/FacebookAI/xlm-roberta-large/blob/main/pytorch_model.bin) and fine tune on our custom dataset for 29 epochs with a learning rate of 5e-05 and a batch size of 42. We chose the learning rate by tuning on the validation set.
|
26 |
|
27 |
### Usage
|
28 |
|