umap library version and embed_umap()

[rogelioamancisidor]

Thanks for sharing your model implementation. I have installed all dependencies as specified in the readme file. However, there is something that it is wrong. In the function embed_umap() there is an argument called transform_seed. It seems like transform_seed was first introduced in umap-learn==0.3.0 and not in umap-learn==0.1.1. It does not help to install umap-learn==0.3.0 because that will trigger an error message since there is a dependency with joblib in sklearn. So sklearn must be an older version, but I am not sure which one. I maneged to run python main.py --model mnist_svhn by replacing UMAPwith TSNE in the function embed_umap.

Another issue is that the code seems to be running very slow, eventhough it runs in a GPU. Any idea what can it be happening? can it be the function embed_umap() that it is triggered at each epoch?

[zihan0822]

Same dependency issue here. For me, when trained on single 3090, it takes round 10 min per epoch for mnist-svhn dataset.