state_dict

torch.nn.Module.state_dict (Python method, in Module)

state_dict(destination=None, prefix='', keep_vars=False)[source]

返回一个包含整个模型状态的字典。包含参数和现在的缓冲器(例如,运行平均值)。键对应着参数和缓冲器的名字。

返回值:

  • 包含整个模型的字典。

a dictionary containing a whole state of the module

例:

>>> module.state_dict().keys()
['bias', 'weight']

torch.optim.Optimizer.state_dict (Python method, in torch.optim)

state_dict()[source]

以字典的形式返回优化器的状态。

包含两个词目:

  • state - 保持当前优化状态的字典,它的内容和优化器类不同。
  • param_groups - 包含所有参数组的字典

torch.optim.lr_scheduler.LambdaLR.state_dict (Python method, in torch.optim)

class torch.optim.lr_scheduler.MultiplicativeLR(optimizer, lr_lambda, last_epoch=-1)[source]

将每个参数组的学习率乘以指定函数中给定的因子。当last_epoch = -1时,设置学习率为初始学习率。

参数:

  • optimizer (Optimizer) – 包裹的优化器
  • lr_lambda (function or list) – A function which computes a multiplicative factor given an integer parameter epoch, or a list of such functions, one for each group in optimizer.param_groups.
  • lr_lambda (function or list) – 给定整数参数epoch计算乘法因子的函数,或者这个函数的列表,在optimizer.param_groups中,每个组都有一个。
  • last_epoch (int) – 最后一个epoch的指数,默认是-1。

例:

>>> lmbda = lambda epoch: 0.95
>>> scheduler = MultiplicativeLR(optimizer, lr_lambda=lmbda)
>>> for epoch in range(100):
>>>     train(...)
>>>     validate(...)
>>>     scheduler.step()

load_state_dict(state_dict)[source]

加载策略状态

参数:

  • state_dict (dict) –策略状态。应该是调用state_dict()返回的对象。

state_dict()[source]

Returns the state of the scheduler as a dict.It contains an entry for every variable in self.__dict__ which is not the optimizer. The learning rate lambda functions will only be saved if they are callable objects and not if they are functions or lambdas.

以字典的形式返回策略的状态。对每个变量它包含self.__dict__中的实体,这不是优化器。如果它们是可以调用的对象的话,学习率lambda函数就保存,如果他们是函数或者lambdas的话就不保存。

承接Matlab、Python和C++的编程,机器学习、计算机视觉的理论实现及辅导,本科和硕士的均可,咸鱼交易,专业回答请走知乎,详谈请联系QQ号757160542,非诚勿扰。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • python中的__dict__和dir()的区别

    Python下一切皆对象,每个对象都有多个属性(attribute),Python对属性有一套统一的管理方案。

    于小勇
  • state_dict详解

    在pytorch中,torch.nn.Module模块中的state_dict变量存放训练过程中需要学习的权重和偏执系数,state_dict作为python的...

    于小勇
  • Python的dict()函数

    版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。 ...

    于小勇
  • CVE-2019-7805漏洞分析

    漏洞摘要 Adobe Acrobat Reader DC中存在一个释放后使用的漏洞,它允许攻击者使用当前用户的权限执行任意代码。 CVE CVE-2019-78...

    周俊辉
  • 树莓派基本配置与Docker的安装

    https://www.raspberrypi.org/downloads/raspbian/

    緣來
  • Python Dict用法

    dict = {"a" : "apple", "b" : "banana", "g" : "grape", "o" : "orange"}

    py3study
  • PyTorch的元学习库:Torchmeta

    Torchmeta是扩展和数据加载器的集合,用于在PyTorch中进行少量学习和元学习。Torchmeta在2019年全球PyTorch夏季黑客马拉松上获得了最...

    代码医生工作室
  • 【重磅】雷鸣对话吴恩达(Andrew Ng):超级大咖深度解析人工智能的发展现状与未来

    【新智元导读】2016年4月14日(周四)21:00 - 22:30,酷我创始人、北大大数据与机器学习中心联合主任雷鸣先生以线上对话的方式,在微信群中与人工智能...

    新智元
  • 3. react-redux

    react-redux是reactjs官方推荐的state管理器。具体的定义我就不说了,因为有很多地方比我说的好,大家可以Google或参照:redux、中文文...

    MasterVin
  • ajax无刷新页面切换,历史记录后退前进解决方案

    我们在工作中常常遇到需要用ajax来显示下一页和上一页,ajax可以不刷新页面进行操作!但是,假如你想通过浏览器的历史记录返回上一页和下一页。那么ajax默认是...

    用户7657330

扫码关注云+社区

领取腾讯云代金券