MCPcopy Create free account
hub / github.com/Meshcapade/difflocks / kl

Function kl

losses/losses.py:96–98  ·  view source on GitHub ↗
(mean, logstd)

Source from the content-addressed store, hash-verified

94 return kl_loss
95
96def kl(mean, logstd):
97 kl = (-0.5 - logstd + 0.5 * mean ** 2 + 0.5 * torch.exp(2 * logstd))
98 return kl
99
100
101

Callers 1

compute_loss_klFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected