tomaarsen HF staff AlexandrosChariton commited on
Commit
4e754b7
1 Parent(s): 67b8354

replaced max with avg in comment (#8)

Browse files

- replaced max with avg in comment (a4b21d1980c45a0499e24294477b447f80b00a11)


Co-authored-by: Alexandros Chariton <AlexandrosChariton@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -120,7 +120,7 @@ encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tenso
120
  with torch.no_grad():
121
  model_output = model(**encoded_input)
122
 
123
- # Perform pooling. In this case, max pooling.
124
  sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
125
 
126
  print("Sentence embeddings:")
 
120
  with torch.no_grad():
121
  model_output = model(**encoded_input)
122
 
123
+ # Perform pooling. In this case, average pooling
124
  sentence_embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
125
 
126
  print("Sentence embeddings:")