matchaoneshot commited on
Commit
1e3a38d
·
verified ·
1 Parent(s): edb41a4

update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -9
README.md CHANGED
@@ -22,17 +22,20 @@ This model was fine-tuned for the task of **Multi-Intent Detection (MID)**, a ty
22
  - **Number of Labels:** 2
23
  - **Fine-tuning Framework:** Hugging Face Transformers
24
 
 
25
  ## Training Configuration
26
 
27
- - **Epochs:** 8
28
- - **Batch Size:** 16
29
- - **Learning Rate:** 2e-5
30
- - **Optimizer:** AdamW
31
- - **Scheduler:** Cosine Learning Rate Scheduler
32
- - **Gradient Accumulation Steps:** 2
33
- - **Mixed Precision Training:** Enabled (FP16)
34
- - **Weight Decay:** 0.01
35
- - **Warmup Ratio:** 10%
 
 
36
 
37
  ## Performance Metrics
38
 
 
22
  - **Number of Labels:** 2
23
  - **Fine-tuning Framework:** Hugging Face Transformers
24
 
25
+
26
  ## Training Configuration
27
 
28
+ - **Training Arguments:**
29
+ - **Learning Rate:** 2e-5
30
+ - **Batch Size (Train):** 16
31
+ - **Batch Size (Eval):** 16
32
+ - **Gradient Accumulation Steps:** 2
33
+ - **Number of Epochs:** 8
34
+ - **Weight Decay:** 0.01
35
+ - **Warmup Ratio:** 10%
36
+ - **Learning Rate Scheduler Type:** Cosine
37
+ - **Mixed Precision Training:** Enabled (FP16)
38
+ - **Logging Steps:** 50
39
 
40
  ## Performance Metrics
41