3分钟
定义损失函数和优化器
我们使用交叉熵作为损失函数，使用带动量的随机梯度下降。
In [5]:
import torch.optim as optim
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)我们使用交叉熵作为损失函数，使用带动量的随机梯度下降。
In [5]:
import torch.optim as optim
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)Copyright © 2013-2025 Tencent Cloud.All Rights Reserved. 腾讯云 版权所有
学员评价