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