fix: add missing module
Browse files- modeling_xlm_roberta.py +1 -0
modeling_xlm_roberta.py
CHANGED
|
@@ -944,6 +944,7 @@ class XLMRobertaForSequenceClassification(XLMRobertaPreTrainedModel):
|
|
| 944 |
)
|
| 945 |
.float()
|
| 946 |
)
|
|
|
|
| 947 |
all_scores.extend(scores.cpu().numpy().tolist())
|
| 948 |
|
| 949 |
if len(all_scores) == 1:
|
|
|
|
| 944 |
)
|
| 945 |
.float()
|
| 946 |
)
|
| 947 |
+
scores = torch.sigmoid(scores)
|
| 948 |
all_scores.extend(scores.cpu().numpy().tolist())
|
| 949 |
|
| 950 |
if len(all_scores) == 1:
|