在遵循《使用地标重新训练参数 UMAP》教程时

嘿，umap 团队！ 首先，非常感谢你们在该库上的所有工作，它非常有用！ 如果能够保留已经处理过的嵌入的映射，同时重新训练 ParametricUMAP，将是非常棒的。 我尝试将此方法用于自己的用例，参考了 umap-learn 中的示例。 然而，在重新训练阶段，每个 epoch 的损失总是为 nan。 我猜测这是由于我自己的设置出现了问题，因此我照搬了示例。 遗憾的是，我得到的结果完全相同。 模型无法成功重新训练。