Skip to content

Commit 3f5aa4e

Browse files
authored
Update BBBConv.py
The variable "sample" is not defined but used in Line 77. I add a default value "True" for it as in other files.
1 parent 16e2462 commit 3f5aa4e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

layers/BBB_LRT/BBBConv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def reset_parameters(self):
5959
self.bias_mu.data.normal_(*self.posterior_mu_initial)
6060
self.bias_rho.data.normal_(*self.posterior_rho_initial)
6161

62-
def forward(self, x):
62+
def forward(self, x, sample=True):
6363

6464
self.W_sigma = torch.log1p(torch.exp(self.W_rho))
6565
if self.use_bias:

0 commit comments

Comments
 (0)