Ouput Variance - Githubissues

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

Ouput Variance #56

Open aunsid opened 4 years ago

aunsid commented 4 years ago

I am trying to replace nLL with the Heteroscedastic uncertainty model using the formula given here which expects the model to output the the predicted mean and variance. I know that the output returns a tuple of predicted mean and KL but is there a way to get the predicted variance?