Data-Science-Notes
Data-Science-Notes copied to clipboard
8.deep-learning/PyTorch_beginner/2.Autograd自动求导.ipynb 勘误
教程最后部分:
只优化分类器
optimizer = optim.SGD(model.fc.parameters(), lr=1e-2, momentum=0.9) 原文应该是: optimizer = optim.SGD(model.parameters(), lr=1e-2, momentum=0.9)