教程最后部分: # 只优化分类器 optimizer = optim.SGD(model.fc.parameters(), lr=1e-2, momentum=0.9) 原文应该是: optimizer = optim.SGD(model.parameters(), lr=1e-2, momentum=0.9)
教程最后部分:
只优化分类器
optimizer = optim.SGD(model.fc.parameters(), lr=1e-2, momentum=0.9)
原文应该是:
optimizer = optim.SGD(model.parameters(), lr=1e-2, momentum=0.9)