2021
03-05
03-05
Pytorch 中的optimizer使用说明
与优化函数相关的部分在torch.optim模块中,其中包含了大部分现在已有的流行的优化方法。如何使用Optimizer要想使用optimizer,需要创建一个optimizer对象,这个对象会保存当前状态,并根据梯度更新参数。怎样构造Optimizer要构造一个Optimizer,需要使用一个用来包含所有参数(Tensor形式)的iterable,把相关参数(如learningrate、weightdecay等)装进去。注意,如果想要使用.cuda()方法来将model移到GPU中,一定要确保这一...
继续阅读 >