【pytorch官方文档学习之六】torch.optim

本系列旨在通过阅读官方pytorch代码熟悉CNN各个框架的实现方式和流程。

【pytorch官方文档学习之六】torch.optim

  • 本文是对官方文档PyTorch: optim的详细注释和个人理解,欢迎交流。
  • learnable parameters的缺点
    本系列的之前几篇文章已经可以做到使用torch.no_grad.data来手动更改可学习参数的tensors来更新模型的权重。但是这种方法对于简单的优化算法,如stochastic gradient descent随机梯度下降尚可,而对于实际中更常见的复杂的优化器optimizer,如AdamAdaGradRMSProp等就显得过于臃肿。
    为了解决这个问题,torch.optim包应运而生。
  • optim
    torch.optim包可实现常用的优化算法。
  • 实例
    以下实例使用torch.nn.Module来定义神经网络模型,使用torch.optim提供的Adam算法优化神经网络模型。
# -*- coding: utf-8 -*-
import torch

# N is batch size; D_in is input dimension;
# H is hidden dimension; D_out is output dimension. 
N, D_in, H, D_out = 64, 1000, 100
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值