import torch from torch.autograd import Function, Variable class ... logits * gt + torch.log(1 + torch.exp(-logits.abs())) loss = loss * w loss = loss.sum() ...
確定! 回上一頁