在用pytorch训练模型时，通常会在遍历epochs的过程中依次用到optimizer.zero_grad(),loss.backward()和optimizer.step()三个函数，如下所示：
model = MyModel()
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.SGD(model.parameters(), lr=0.001, momentumpytorch训练是怎样递进的
猜你喜欢
转载自blog.csdn.net/juluwangriyue/article/details/120961696
今日推荐
周排行