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?
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 calledtransform_seed. It seems liketransform_seedwas first introduced inumap-learn==0.3.0and not inumap-learn==0.1.1. It does not help to installumap-learn==0.3.0because that will trigger an error message since there is a dependency withjoblibinsklearn. So sklearn must be an older version, but I am not sure which one. I maneged to runpython main.py --model mnist_svhnby replacingUMAPwithTSNEin the functionembed_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?