search

Atcold / NYU-DLSP20

NYU Deep Learning Spring 2020
https://atcold.github.io/NYU-DLSP20/
Other
6.67k stars 2.22k forks source link

[11-VAE.ipnb] TSNE fit_transform() happens a error with cuda #776

Open norihiro-ito opened 3 years ago

norihiro-ito commented 3 years ago

Following line happens a coversion error in the 16th cell. with CUDA environment. It causes to set cuda type tensor to TSNE,fit_transform().

E.append(TSNE(n_components=2).fit_transform(X[-1]))

I passed to add ,cpu() calling.

E.append(TSNE(n_components=2).fit_transform(X[-1].cpu()))

Atcold commented 3 years ago

It looks good. @norihiro-ito, do you think you can send a pull request with such improvement? Otherwise @ritchieng can do it for you.

ritchieng commented 3 years ago

Let me double check. I'll push the fix and close the issue once I'm done.