GAN-详解BCELoss和BCEWithLogitsLoss
一、BCELoss()生成对抗网络的所使用到的loss函数BCELoss和BCEWithLogitsLoss其中BCELoss的公式为:其中y是target,x是模型输出的值。二、例子import torchfrom torch import autogradfrom torch import nnimport mathinput = autograd.Variable(torch.tensor([[ 1.9072, 1.1079, 1.4906], [-0.658
原创
2022-02-17 17:37:05 ·
2730 阅读 ·
0 评论