wassname/attentive-neural-processes
1
0
Fork 0
mirror of https://github.com/wassname/attentive-neural-processes.git synced 2026-06-27 16:59:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #5

Browse Source
This commit is contained in:
Mike Clark
2022-03-17 21:51:55 +00:00
committed by GitHub
parent 016272a077
commit eea4a7e35a
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
neural_processes/utils.py
+3 -1
View File
@@ -172,8 +172,10 @@ def kl_loss_var(prior_mu, log_var_prior, post_mu, log_var_post):
"""
var_ratio_log = log_var_post - log_var_prior
t1 = (post_mu - prior_mu) ** 2 / log_var_prior.exp()
kl_div = (
(var_ratio_log.exp() + (post_mu - prior_mu) ** 2) / log_var_prior.exp()
var_ratio_log.exp()
+ t1
- 1.0
- var_ratio_log
)