擅长:python、mysql、java
<p>您可能需要使用<code>torch.nn.BCEWithLogitsLoss()</code>,替换Sigmoid和BCELoss函数。在</p>
<p>从<a href="http://pytorch.org/docs/master/nn.html#torch.nn.BCEWithLogitsLoss" rel="nofollow noreferrer">docs</a>中摘录的内容告诉您为什么使用这种丢失函数实现总是更好。在</p>
<blockquote>
<p>This loss combines a Sigmoid layer and the BCELoss in one single class. This version is more numerically stable than using a plain Sigmoid followed by a BCELoss as, by combining the operations into one layer, we take advantage of the log-sum-exp trick for numerical stability.</p>
</blockquote>