search

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

Bayesian UNet #80

Closed amsinha9 closed 3 years ago

amsinha9 commented 3 years ago

Do you have any examples we could use for a Bayesian UNet (3D)? I'm really interested in using blitz for some research, so this would be helpful!

piEsposito commented 3 years ago

You can try to replicate the UNet replacing the Conv Layers for Bayesian ones, or just putting some Bayesian layers on the top of your network.