Converting Neural Network to pyro Bayesian Neural Networ

Hi! I want to convert densenet to bayesian network. I don’t want to manually convert each layer to pyro. Is there an easy to do it?

Hi, you probably want to start with the documentation and tutorial on pyro.nn.PyroModule. pyro.nn.to_pyro_module_ is probably closest to what you have in mind; see the docstring there for a code snippet.

1 Like