OML-Team/open-metric-learning

Avoid storing full distance matrix during validation

Closed this issue · 1 comments

We don't really need to store full QxG matrix due to its weight.

We can compute matmul + top_k in a batchwise rejime and then clip the second dimension, so the resulted matrix is QxC, where C >> k (where k is from map@k, precision@k, recall@k)

now it's a part of: #522