Open longcwang3 opened 1 year ago
For calculate_kl, I am assuming mu_q refers to the variational posterior and mu_p refers to the prior. https://github.com/kumar-shridhar/PyTorch-BayesianCNN/blob/d93bad543c3226cd0fe05c0cb0ba033c41b3caa6/metrics.py#L27-L29
calculate_kl
mu_q
mu_p
While in the following codes, prior_mu is input as mu_q. https://github.com/kumar-shridhar/PyTorch-BayesianCNN/blob/d93bad543c3226cd0fe05c0cb0ba033c41b3caa6/layers/BBB_LRT/BBBLinear.py#L75-L79
prior_mu
Could you check this?
For
calculate_kl
, I am assumingmu_q
refers to the variational posterior andmu_p
refers to the prior. https://github.com/kumar-shridhar/PyTorch-BayesianCNN/blob/d93bad543c3226cd0fe05c0cb0ba033c41b3caa6/metrics.py#L27-L29While in the following codes,
prior_mu
is input asmu_q
. https://github.com/kumar-shridhar/PyTorch-BayesianCNN/blob/d93bad543c3226cd0fe05c0cb0ba033c41b3caa6/layers/BBB_LRT/BBBLinear.py#L75-L79Could you check this?