Update README.md
#1
by NextAITeam - opened
README.md
CHANGED
|
@@ -67,8 +67,7 @@ widget:
|
|
| 67 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 68 |
import torch
|
| 69 |
|
| 70 |
-
# مسیر بارگذاری مدل از هاب
|
| 71 |
-
# جایگزین کنید با آدرس دقیق مدل در اکانت خود: "username/NEXARA_model"
|
| 72 |
MODEL_NAME = "MTE313/NEXARA_model"
|
| 73 |
|
| 74 |
# بارگذاری توکنایزر و مدل
|
|
@@ -133,8 +132,8 @@ for sentence in test_sentences:
|
|
| 133 |
مدل با پارامترهای زیر بر روی دیتاست آموزشی Fine-Tune شده است:
|
| 134 |
|
| 135 |
- **Optimizer:** AdamW
|
| 136 |
-
- **Learning Rate:** 2e-5
|
| 137 |
-
- **Batch Size:**
|
| 138 |
- **Epochs:** 3
|
| 139 |
- **Warmup Steps:** 500
|
| 140 |
- **Weight Decay:** 0.01
|
|
|
|
| 67 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 68 |
import torch
|
| 69 |
|
| 70 |
+
# مسیر بارگذاری مدل از هاب hugging face
|
|
|
|
| 71 |
MODEL_NAME = "MTE313/NEXARA_model"
|
| 72 |
|
| 73 |
# بارگذاری توکنایزر و مدل
|
|
|
|
| 132 |
مدل با پارامترهای زیر بر روی دیتاست آموزشی Fine-Tune شده است:
|
| 133 |
|
| 134 |
- **Optimizer:** AdamW
|
| 135 |
+
- **Learning Rate:** 2e-5
|
| 136 |
+
- **Batch Size:** 8
|
| 137 |
- **Epochs:** 3
|
| 138 |
- **Warmup Steps:** 500
|
| 139 |
- **Weight Decay:** 0.01
|