From 3f5aa4e4aa3508e2febd3072a638dd87ef340438 Mon Sep 17 00:00:00 2001 From: lan-qing Date: Thu, 24 Sep 2020 13:31:05 +0800 Subject: [PATCH] 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. --- layers/BBB_LRT/BBBConv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/layers/BBB_LRT/BBBConv.py b/layers/BBB_LRT/BBBConv.py index 5b2f2b5e..c4a7d162 100755 --- a/layers/BBB_LRT/BBBConv.py +++ b/layers/BBB_LRT/BBBConv.py @@ -59,7 +59,7 @@ def reset_parameters(self): self.bias_mu.data.normal_(*self.posterior_mu_initial) self.bias_rho.data.normal_(*self.posterior_rho_initial) - def forward(self, x): + def forward(self, x, sample=True): self.W_sigma = torch.log1p(torch.exp(self.W_rho)) if self.use_bias: