Skip to content
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

get_targets部分生成较慢 #17

Open
Kxitinga opened this issue May 8, 2022 · 2 comments
Open

get_targets部分生成较慢 #17

Kxitinga opened this issue May 8, 2022 · 2 comments

Comments

@Kxitinga
Copy link

Kxitinga commented May 8, 2022

image
您好,我按照您的办法,在get_targets部分生成sigma分别为0.02 0.08 0.003 的npz文件,但是生成过程较慢,平均6分钟一个,随机无序生成,一个2000个左右的文件,请问这是正常的吗?

@zhaofang0627
Copy link
Owner

zhaofang0627 commented May 9, 2022

这里用的sklearn里的最近邻算法,速度是会比较慢。可以考虑使用基于GPU的算法库来加速,例如faiss

@Kxitinga
Copy link
Author

Kxitinga commented May 9, 2022

好的,感谢回复

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants