Request GPU version of re_rank function
jpainam opened this issue · 3 comments
jpainam commented
Hi!
When using multi query evaluation in Market1501, this
original_dist = np.concatenate(
[np.concatenate([q_q_dist, q_g_dist], axis=1),
np.concatenate([q_g_dist.T, g_g_dist], axis=1)],
axis=0)
shows MemoryError even in a 32GB memory window PC. This line of code should run on the GPU
Kindly provide a re_rank_gpu version of this function for very large evaluation
Thank you
PayneYong commented
jpainam commented
No, i didn't solve. I decided not to include re-ranking at all since the author of the implementation rarely replies
PayneYong commented
OK