MCPcopy Create free account
hub / github.com/Zhiyuan-R/Tiger-Diffusion / KLLoss

Class KLLoss

modules/loss.py:8–10  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

6
7
8class KLLoss(nn.Module):
9 def forward(self, x, y):
10 return F.kl_loss(x, y)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected