利用Pytorch对CIFAR10数据集分类（三）

定义损失函数和优化器

代码如下：

net = Net()
import numpy as numpy 
from torch.autograd import Variable
# Define loss (Cross-Entropy)
import torch.optim as optim

criterion = nn.CrossEntropyLoss()
# SGD with momentum
#optimizer = optim.SGD(net.parameters(), lr=0.005, momentum=0.9)
optimizer = optim.Adam(net.parameters(),lr = 0.005)

其中

net = Net()

该语句为网络定义一个对象，上面我们所建的网络是一个类。

criterion = nn.CrossEntropyLoss()

CrossEntropyLoss交叉商损失函数
交叉熵的理解参考
https://blog.youkuaiyun.com/xg123321123/article/details/52864830

#optimizer = optim.