Merge pull request #58 from lan-qing/master

Update BBBConv.py

Author: kumar-shridhar

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: