-
Notifications
You must be signed in to change notification settings - Fork 558
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Error occurred when running hard negative mining #1226
Comments
Hello, @wongbokhe1. You can remove |
Thanks a lot! After solving the faiss-gpu problem, a new error related to 'AbsEmbedder.py' occur: |
Please refer to: #1187 (comment) |
python hn_mine.py --model_name_or_path /data/bxhuang/model_folder/xiaobu-embedding-v2 --input_file /data/bxhuang/Reranking/train_data/text_pairs.json --output_file /data/bxhuang/Reranking/train_data/text_pairs_HN.json --range_for_sampling 50-80 --negative_number 15 --use_gpu_for_searching
Traceback (most recent call last):
File "/data/bxhuang/FlagEmbedding/scripts/hn_mine.py", line 116, in
find_knn_neg(model,
File "/data/bxhuang/FlagEmbedding/scripts/hn_mine.py", line 83, in find_knn_neg
index = create_index(p_vecs, use_gpu=use_gpu)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/bxhuang/FlagEmbedding/scripts/hn_mine.py", line 29, in create_index
co = faiss.GpuMultipleClonerOptions()
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: module 'faiss' has no attribute 'GpuMultipleClonerOptions'
The text was updated successfully, but these errors were encountered: