Update README.md
Browse files
README.md
CHANGED
|
@@ -43,4 +43,24 @@ with torch.no_grad():
|
|
| 43 |
embeddings = last_hidden_state[:, 0]
|
| 44 |
print("embeddings:")
|
| 45 |
print(embeddings)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
```
|
|
|
|
| 43 |
embeddings = last_hidden_state[:, 0]
|
| 44 |
print("embeddings:")
|
| 45 |
print(embeddings)
|
| 46 |
+
```
|
| 47 |
+
|
| 48 |
+
## Contact
|
| 49 |
+
If you have any question or suggestion related to this project, feel free to open an issue or pull request.
|
| 50 |
+
You also can email Tong Wu(townswu@tencent.com).
|
| 51 |
+
|
| 52 |
+
|
| 53 |
+
## Citation
|
| 54 |
+
|
| 55 |
+
If you find our work helpful for your research, please consider citing the following BibTeX entry:
|
| 56 |
+
|
| 57 |
+
```
|
| 58 |
+
|
| 59 |
+
@article{wu2023towards,
|
| 60 |
+
title={Towards Robust Text Retrieval with Progressive Learning},
|
| 61 |
+
author={Wu, Tong and Qin, Yulei and Zhang, Enwei and Xu, Zihan and Gao, Yuting and Li, Ke and Sun, Xing},
|
| 62 |
+
journal={arXiv preprint arXiv:2311.11691},
|
| 63 |
+
year={2023}
|
| 64 |
+
}
|
| 65 |
+
|
| 66 |
```
|