然后,你可以设置optimizer的参数选项,比如学习率,权重衰减,等等。 例子: optimizer = optim.SGD(model.parameters(), lr = 0.01, momentum=0.9) optimizer = ...
確定! 回上一頁