Added sample default parameter to forward method

kumar-shridhar / PyTorch-BayesianCNN

Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.

MIT License

1.43k stars 324 forks source link

Added sample default parameter to forward method #46

Closed tuero closed 4 years ago

tuero commented 4 years ago

BBB_MCMF_LRT/BBBconv.py layer did not have the 'sample' variable defined in the forward method, and so would crash when called. The sample variable has been added as a default parameter to the forward method, matching expected behaviour of the other layers.

Piyush-555 commented 4 years ago

@tuero Thanks for the bug fix!