Sampling values directly on GPU, instead of copying from CPU to GPU. …

kumar-shridhar / PyTorch-BayesianCNN

Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.

MIT License

1.42k stars 323 forks source link

Closed sarahdoerr closed 5 years ago

sarahdoerr commented 5 years ago

…7 times faster.

kumar-shridhar commented 5 years ago

Hi @sarahdoerr

Makes sense. I am incorporating these changes and will test. Thanks.