jupyterjazz
commited on
Commit
•
b20a611
1
Parent(s):
f2933ac
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.
|