IntelLabs / bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch
BSD 3-Clause "New" or "Revised" License
518 stars 72 forks source link

Added support for arbitrary kernel sizes for Bayesian Conv layers #18

Closed msubedar closed 1 year ago