关于扩大训练数据规模的问题

[TMaysGGS]

作者你好，
非常感谢论文和开源代码，我想尝试通过扩大训练数据集的规模（同时提升人脸识别模型的feature维度为512）来试图提升最终聚类模型的性能。但当我尝试使用了一个100万规模的训练集（自制的正好一百万数据）时，卡在了建立knn这一步骤。我使用3090显卡，但我发现这一步骤只占用2G显存，且过去了3周时间依然无进展。请问作者有没有什么加速这一步骤的建议？感谢。