DBence17

Nan during GPU training

6

Dear Developers, I'm encountering an issue during Parametric UMAP training on GPU: the loss becomes NaN, even when using the official demo notebooks without any modifications. My setup: - GPU:...