DaehanKim / vgae_pytorch

This repository implements variational graph auto encoder by Thomas Kipf.
MIT License
385 stars 73 forks source link

a small problem #2

Closed chenbofeng123 closed 4 years ago

chenbofeng123 commented 4 years ago

kl_divergence = 0.5/ A_pred.size(0) (1 + 2model.logstd - model.mean2 - torch.exp(model.logstd)).sum(1).mean() torch.exp(model.logstd)2 ?

DaehanKim commented 4 years ago

Dear @chenbofeng123, Thank you for your correction! I'll update it.