import torch
import torch.nn as nn
model = nn.Linear(10, 1)
criterion = nn.BCEWithLogitsLoss()
x = torch.randn(16, 10)
y
pytorch测试loss的简易方法
猜你喜欢
转载自blog.csdn.net/dragon_T1985/article/details/121032614
今日推荐
周排行