最近使用了pytorch的KL散度计算模块,发现loss是负值,例如下面的代码。
import torch
KL_criterion = torch.nn.KLDivLoss(size_average=False)
a = torch.tensor([0.1, 0.2, 0.3, 0.4])
最近使用了pytorch的KL散度计算模块,发现loss是负值,例如下面的代码。
import torch
KL_criterion = torch.nn.KLDivLoss(size_average=False)
a = torch.tensor([0.1, 0.2, 0.3, 0.4])