tyxe/bnn.py
def named_pytorch_parameters(self, *input_data):
        """Equivalent of the named_parameters method of an nn.Module. Ensures that prior and guide are run once to
        initialize all pyro parameters. Those are then collected and returned via the trace poutine."""
        model_trace = poutine.trace(self.net, param_only=True).get_trace(*input_data)
        guide_trace = poutine.trace(self.net_guide, param_only=True).get_trace(*input_data)
        for name, msg in itertools.chain(model_trace.nodes.items(), guide_trace.nodes.items()):
            yield name, msg["value"].unconstrained()
Equivalent of the named_parameters method of an nn.Module. Ensures that prior and guide are run once to
        initialize all pyro parameters. Those are then collected and returned via the trace poutine.
Equivalent of the named_parameters method of an nn.Module. Ensures that prior and guide are run once to
initialize all pyro parameters. Those are then collected and returned via the trace poutine.
Equivalent of the named_parameters method of an nn . Module . Ensures that prior and guide are run once to initialize all pyro parameters . Those are then collected and returned via the trace poutine .
def named_pytorch_parameters(self, *input_data):
        model_trace = poutine.trace(self.net, param_only=True).get_trace(*input_data)
        guide_trace = poutine.trace(self.net_guide, param_only=True).get_trace(*input_data)
        for name, msg in itertools.chain(model_trace.nodes.items(), guide_trace.nodes.items()):
            yield name, msg["value"].unconstrained()
def named_pytorch_parameters ( self , * input_data ) : model_trace = poutine . trace ( self . net , param_only = True ) . get_trace ( * input_data ) guide_trace = poutine . trace ( self . net_guide , param_only = True ) . get_trace ( * input_data ) for name , msg in itertools . chain ( model_trace . nodes . items ( ) , guide_trace . nodes . items ( ) ) : yield name , msg [ "value" ] . unconstrained ( )
Equivalent of the named_parameters method of an nn.Module.
Equivalent of the named_parameters method of an nn . Module .
"""Equivalent of the named_parameters method of an nn.Module. Ensures that prior and guide are run once to
        initialize all pyro parameters. Those are then collected and returned via the trace poutine."""
self