mirror of
https://github.com/facebookresearch/faiss.git
synced 2026-10-12 02:13:36 +00:00
Set NN Descent Metric From CAGRA Params (#4540)
Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/4540 Reviewed By: mnorris11 Differential Revision: D80734549 Pulled By: junjieqi fbshipit-source-id: 0b19e3e6ed05731e7076fb230f45503b36a3faac
This commit is contained in:
committed by
Facebook GitHub Bot
parent
5c61ed8c94
commit
fbbb290a57
@@ -193,6 +193,7 @@ void CuvsCagra<data_t>::train(idx_t n, const data_t* x) {
|
||||
cuvs::neighbors::cagra::graph_build_params::nn_descent_params
|
||||
graph_build_params(index_params_.intermediate_graph_degree);
|
||||
graph_build_params.max_iterations = nn_descent_niter_;
|
||||
graph_build_params.metric = metricFaissToCuvs(metric_, false);
|
||||
index_params_.graph_build_params = graph_build_params;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user