We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 44124b8 commit 7d3f8e3Copy full SHA for 7d3f8e3
pyro/nn/auto_reg_nn.py
@@ -19,7 +19,7 @@ class MaskedLinear(nn.Linear):
19
:param mask: the mask to apply to the in_features x out_features weight matrix
20
:type mask: torch.autograd.Variable
21
:param bias: whether or not `MaskedLinear` should include a bias term. defaults to `True`
22
- :type bias bool:
+ :type bias: bool
23
"""
24
def __init__(self, in_features, out_features, mask, bias=True):
25
super(MaskedLinear, self).__init__(in_features, out_features, bias)
0 commit comments