bugfix - zeroed bias need to inherit device

piEsposito / blitz-bayesian-deep-learning

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.

GNU General Public License v3.0

918 stars 107 forks source link

Closed Lkruitwagen closed 2 years ago

Lkruitwagen commented 2 years ago

Addresses #109

n.b. I don't think Object classes inherit a self.device attribute? So I've set the zero tensor to acquire device from x.

piEsposito commented 2 years ago

Closes #109.