converting prob into a cuda variable

timbmg / Sentence-VAE

PyTorch Re-Implementation of "Generating Sentences from a Continuous Space" by Bowman et al 2015 https://arxiv.org/abs/1511.06349

580 stars 152 forks source link

Closed dhanajitb closed 5 years ago

dhanajitb commented 5 years ago

Simply converting the prob into a cuda variable to make it work in GPU.