See related issues: - https://github.com/KevinMusgrave/pytorch-metric-learning/issues/179#issuecomment-1501168068 - https://github.com/KevinMusgrave/pytorch-metric-learning/discussions/606#discussioncomment-5542877 - https://github.com/KevinMusgrave/pytorch-metric-learning/issues/600#issuecomment-1483894037
