Pytorch 基础操作第二部分

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

龙良曲 相关笔记

Pytorch学习笔记

GPU

刘二大人

model = Net()
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)

inputs, target = data
inputs, target = inputs.to(device), target.to(device)

龙良曲

device = torch.device('cuda:0')
net = MLP().to(device)
criteon = nn.CrossEntropyLoss().to(device)
data, target = data.to(device), target.to(device)

$Softmax$

$S(y_i)={\sum }_j$