See: https://github.com/KevinMusgrave/pytorch-metric-learning/issues/374#issuecomment-952745778