... loss criterion = FocalLoss(gamma=0). Examples. For Binary-classification batch_size = 10 m = torch.nn.Sigmoid() logits = torch.randn(batch_size) target = torch.
確定! 回上一頁