ilya-kolchinsky commited on
Commit
9800a1c
·
verified ·
1 Parent(s): afca3cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -107,7 +107,7 @@ def route_prompt(prompt: str) -> str:
107
  ### Model
108
  - **Backbone encoder:** microsoft/deberta-v3-base
109
  - Mask-aware **mean pooling** over token embeddings + **LayerNorm**
110
- - **Regression head:** Dropout(0.1) → Linear → ReLU → Linear → Sigmoid
111
  - **Max input length:** 512 tokens
112
  - The model outputs a bounded score in [0, 1]. In the examples below, the score is read from `outputs.logits` (shape `[batch, 1]`).
113
 
 
107
  ### Model
108
  - **Backbone encoder:** microsoft/deberta-v3-base
109
  - Mask-aware **mean pooling** over token embeddings + **LayerNorm**
110
+ - **Regression head:** Linear → ReLU → Linear → Sigmoid
111
  - **Max input length:** 512 tokens
112
  - The model outputs a bounded score in [0, 1]. In the examples below, the score is read from `outputs.logits` (shape `[batch, 1]`).
113