matthewleechen commited on
Commit
f40382c
·
verified ·
1 Parent(s): b0f0bf1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 21. We chose the learning rate by tuning on the validation set.
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