Update README.md
Browse files
README.md
CHANGED
|
@@ -46,7 +46,6 @@ This model is a fine-tuned version of intfloat/multilingual-e5-large for documen
|
|
| 46 |
- Base model: intfloat/multilingual-e5-large
|
| 47 |
- Task: Document text classification
|
| 48 |
- Languages: Multilingual
|
| 49 |
-
- License: MIT
|
| 50 |
|
| 51 |
## Training data
|
| 52 |
- Dataset: DocLayNet-base
|
|
@@ -67,6 +66,7 @@ This model is a fine-tuned version of intfloat/multilingual-e5-large for documen
|
|
| 67 |
Trained on single gpu for 2 epochs for apx. 20 minutes.
|
| 68 |
|
| 69 |
hyperparameters:
|
|
|
|
| 70 |
{
|
| 71 |
'batch_size': 8,
|
| 72 |
'num_epochs': 10,
|
|
@@ -78,7 +78,7 @@ hyperparameters:
|
|
| 78 |
'optimizer': 'AdamW',
|
| 79 |
'scheduler': 'cosine_with_warmup'
|
| 80 |
}
|
| 81 |
-
|
| 82 |
## Evaluation results
|
| 83 |
Test Loss: 0.5192, Test Acc: 0.9719
|
| 84 |
|
|
|
|
| 46 |
- Base model: intfloat/multilingual-e5-large
|
| 47 |
- Task: Document text classification
|
| 48 |
- Languages: Multilingual
|
|
|
|
| 49 |
|
| 50 |
## Training data
|
| 51 |
- Dataset: DocLayNet-base
|
|
|
|
| 66 |
Trained on single gpu for 2 epochs for apx. 20 minutes.
|
| 67 |
|
| 68 |
hyperparameters:
|
| 69 |
+
```python
|
| 70 |
{
|
| 71 |
'batch_size': 8,
|
| 72 |
'num_epochs': 10,
|
|
|
|
| 78 |
'optimizer': 'AdamW',
|
| 79 |
'scheduler': 'cosine_with_warmup'
|
| 80 |
}
|
| 81 |
+
```
|
| 82 |
## Evaluation results
|
| 83 |
Test Loss: 0.5192, Test Acc: 0.9719
|
| 84 |
|