VAE-CVAE-MNIST icon indicating copy to clipboard operation
VAE-CVAE-MNIST copied to clipboard

Published 20 hours ago • verified publisher icon timbmg

Fixes for pytorch 0.4 and running with a GPU

Open dmsj opened this issue 6 years ago • 0 comments

Thanks for the nice tutorial. I ran into some warnings / errors running this with pytorch 0.4.0, so I made quick fixes. If they're helpful to you - feel free to take them. Results look the same as posted previously.

Changes:

  • replaced calls to .data[0] with .item() per pytorch recommendation
  • some changes to work if cuda is available
  • runs on Ubuntu 16.04 with CUDA 9.0 and macOS, both with Pytorch 0.4.0

dmsj avatar Jul 17 '18 16:07 dmsj