outputs = model(inputs) outputs = torch.sigmoid(outputs) # 添加这一行 preds = (outputs > 0.5).float()
outputs = model(inputs)
outputs = torch.sigmoid(outputs) # 添加这一行
preds = (outputs > 0.5).float()