Update README.md
Browse filesFixed the variable name of the final output
README.md
CHANGED
@@ -41,7 +41,7 @@ outputs = model(input_ids=inputs["input_ids"], attention_mask=inputs["attention_
|
|
41 |
# Normalize embeddings for retrieval:
|
42 |
final_output = outputs[0] / outputs[0].norm(dim=-1, keepdim=True)
|
43 |
final_output = final_output.cpu().detach().numpy()
|
44 |
-
print("sequence_output: ",
|
45 |
```
|
46 |
|
47 |
### Extracting Video Embeddings:
|
|
|
41 |
# Normalize embeddings for retrieval:
|
42 |
final_output = outputs[0] / outputs[0].norm(dim=-1, keepdim=True)
|
43 |
final_output = final_output.cpu().detach().numpy()
|
44 |
+
print("sequence_output: ", final_output)
|
45 |
```
|
46 |
|
47 |
### Extracting Video Embeddings:
|