Merge pull request #58 from lan-qing/master

Update BBBConv.py
kumar-shridhar · Sep 27, 2020 · 6ece34b · 6ece34b
2 parents 16e2462 + 3f5aa4e
commit 6ece34b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: