jupyterjazz commited on
Commit
b20a611
1 Parent(s): f2933ac

Update modeling_xlm_roberta.py

Browse files
Files changed (1) hide show
  1. 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 False):
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.