Anatomy of PPl

guide_trace_linear_regression.py

{'b': {'args': (),
       'fn': Normal(loc: 0.0, scale: 1.0),
       'name': 'b',
       'type': 'sample',
       'value': tensor(-1.1897, grad_fn=<AddBackward0>)},
 'b_loc': {'args': (tensor(0.), Real()),
           'fn': <function param.<locals>.fn at 0x127388950>,
           'name': 'b_loc',
           'type': 'param',
           'value': tensor(0., requires_grad=True)},
 'b_scale': {'args': (tensor(1.), GreaterThan(lower_bound=0.0)),
             'fn': <function param.<locals>.fn at 0x1273889d8>,
             'name': 'b_scale',
             'type': 'param',
             'value': tensor(1., grad_fn=<AddBackward0>)},
 'w': {'args': (),
       'fn': Normal(loc: 1.0, scale: 1.0),
       'name': 'w',
       'type': 'sample',
       'value': tensor(-0.4318, grad_fn=<AddBackward0>)},
 'w_loc': {'args': (tensor(1.), Real()),
           'fn': <function param.<locals>.fn at 0x127388840>,
           'name': 'w_loc',
           'type': 'param',
           'value': tensor(1., requires_grad=True)},
 'w_scale': {'args': (tensor(1.), GreaterThan(lower_bound=0.0)),
             'fn': <function param.<locals>.fn at 0x1273888c8>,
             'name': 'w_scale',
             'type': 'param',
             'value': tensor(1., grad_fn=<AddBackward0>)}}

model_trace.py

{'b': {'args': (),
       'fn': Normal(loc: 0.5, scale: 1.0),
       'name': 'b',
       'type': 'sample',
       'value': tensor(0.5519)},
 'obs': {'args': (),
         'fn': Normal(loc: torch.Size([2]), scale: torch.Size([2])),
         'name': 'obs',
         'type': 'sample',
         'value': tensor([0.9150, 1.3150])},
 'w': {'args': (),
       'fn': Normal(loc: 0.0, scale: 1.0),
       'name': 'w',
       'type': 'sample',
       'value': tensor(1.3068)}}

replay_trace.py

{'b': {'args': (),
       'fn': Normal(loc: 0.0, scale: 1.0),
       'name': 'b',
       'type': 'sample',
       'value': tensor(-2.0570, grad_fn=<AddBackward0>)},
 'b_loc': {'args': (tensor(0.), Real()),
           'fn': <function param.<locals>.fn at 0x11f636950>,
           'name': 'b_loc',
           'type': 'param',
           'value': tensor(0., requires_grad=True)},
 'b_scale': {'args': (tensor(1.), GreaterThan(lower_bound=0.0)),
             'fn': <function param.<locals>.fn at 0x11f6369d8>,
             'name': 'b_scale',
             'type': 'param',
             'value': tensor(1., grad_fn=<AddBackward0>)},
 'w': {'args': (),
       'fn': Normal(loc: 1.0, scale: 1.0),
       'name': 'w',
       'type': 'sample',
       'value': tensor(2.8532, grad_fn=<AddBackward0>)},
 'w_loc': {'args': (tensor(1.), Real()),
           'fn': <function param.<locals>.fn at 0x11f636840>,
           'name': 'w_loc',
           'type': 'param',
           'value': tensor(1., requires_grad=True)},
 'w_scale': {'args': (tensor(1.), GreaterThan(lower_bound=0.0)),
             'fn': <function param.<locals>.fn at 0x11f6368c8>,
             'name': 'w_scale',
             'type': 'param',
             'value': tensor(1., grad_fn=<AddBackward0>)}}
{'b': {'args': (),
       'fn': Normal(loc: 0.5, scale: 1.0),
       'name': 'b',
       'type': 'sample',
       'value': tensor(-2.0570, grad_fn=<AddBackward0>)},
 'obs': {'args': (),
         'fn': Normal(loc: torch.Size([2]), scale: torch.Size([2])),
         'name': 'obs',
         'type': 'sample',
         'value': tensor([0.9051, 1.3051])},
 'w': {'args': (),
       'fn': Normal(loc: 0.0, scale: 1.0),
       'name': 'w',
       'type': 'sample',
       'value': tensor(2.8532, grad_fn=<AddBackward0>)}}