hyp1231 commited on
Commit
e8b143c
1 Parent(s): 503d307

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -56,8 +56,8 @@ with torch.no_grad():
56
  embeddings = model(**inputs, return_dict=True).last_hidden_state[:, 0]
57
  embeddings = embeddings / embeddings.norm(dim=1, keepdim=True)
58
 
59
- print(embeddings[0] @ embeddings[1]) # tensor(0.8564)
60
- print(embeddings[0] @ embeddings[2]) # tensor(0.5741)
61
  ```
62
 
63
  ## Citation
 
56
  embeddings = model(**inputs, return_dict=True).last_hidden_state[:, 0]
57
  embeddings = embeddings / embeddings.norm(dim=1, keepdim=True)
58
 
59
+ print(embeddings[0] @ embeddings[1]) # tensor(0.8243)
60
+ print(embeddings[0] @ embeddings[2]) # tensor(0.5814)
61
  ```
62
 
63
  ## Citation