其实就在前面反向传播那里,加了几行,
定义tudui之后,
optim = torch.optim.SGD(tudui.parameters(),lr=0.01)
#随机梯度下降,lr学习率,太大太跳脱,太小训练的慢,一般先大后小
在循环中最后
result_loss = loss(outputs, targets)
optim.zero_grad() #每轮梯度清0
result_loss.backward()
optim.step()
但是我发现我在debug时候没有tudui这个文件夹,看不到梯度下降,挖坑,等我学明白vscode的debug回来写