Update modeling_xlm_roberta.py
Browse files- modeling_xlm_roberta.py +1 -1
modeling_xlm_roberta.py
CHANGED
|
@@ -499,7 +499,7 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
|
|
| 499 |
Overwrites any setting from convert_to_numpy
|
| 500 |
device(`torch.device`, *optional*, defaults to None):
|
| 501 |
Which torch.device to use for the computation
|
| 502 |
-
normalize_embeddings(`bool`, *optional*, defaults to
|
| 503 |
If set to true, returned vectors will have length 1. In that case, the
|
| 504 |
faster dot-product (util.dot_score) instead of cosine similarity can
|
| 505 |
be used.
|
|
|
|
| 499 |
Overwrites any setting from convert_to_numpy
|
| 500 |
device(`torch.device`, *optional*, defaults to None):
|
| 501 |
Which torch.device to use for the computation
|
| 502 |
+
normalize_embeddings(`bool`, *optional*, defaults to True):
|
| 503 |
If set to true, returned vectors will have length 1. In that case, the
|
| 504 |
faster dot-product (util.dot_score) instead of cosine similarity can
|
| 505 |
be used.
|