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 16e2462 commit 3f5aa4eCopy full SHA for 3f5aa4e
layers/BBB_LRT/BBBConv.py
@@ -59,7 +59,7 @@ def reset_parameters(self):
59
self.bias_mu.data.normal_(*self.posterior_mu_initial)
60
self.bias_rho.data.normal_(*self.posterior_rho_initial)
61
62
- def forward(self, x):
+ def forward(self, x, sample=True):
63
64
self.W_sigma = torch.log1p(torch.exp(self.W_rho))
65
if self.use_bias:
0 commit comments