Autoencoder doesn't work on gpu

Open ViralBShah opened this issue 7 years ago • 1 comments

Works on CPU but not GPU. Gives ReadOnlyMemoryError on cyclops with gpu.

https://github.com/aviks/mltraining/blob/master/10.%20MNIST%20Autoencoder.ipynb

cc @mikeinnes

May 03 '18 03:05 ViralBShah

None of the notebooks here work on GPU. I am using released versions of everything on 0.6.2.

May 03 '18 04:05 ViralBShah