这一部分比较简单,就直接上代码了
code实现
import torch.nn as nn
import torch.optim as optim
criterion=nn.CrossEntropyloss() #这里使用的是交叉熵损失
optimizer=optim.SGD(net.parameters(),lr=0.001,momentum=0.9) #随机梯度下降
这一部分比较简单,就直接上代码了
code实现
import torch.nn as nn
import torch.optim as optim
criterion=nn.CrossEntropyloss() #这里使用的是交叉熵损失
optimizer=optim.SGD(net.parameters(),lr=0.001,momentum=0.9) #随机梯度下降