pytorch的state与param_group的区别与联系

之前看到很多文章对param_group进行解释,其实它就是一个长度为1的列表,含有的唯一一个元素是字典类型,其中包括网络模型参数和一些优化器超参数,并不包含状态。

而state,Optimizer中的state是一个defaultdict类型,其中的值默认是字典,一开始是空的,在其自己定义或者官方的优化器中,在step()函数中才将其重新写进去,一般它的键就是网络参数张量,值是一个字典,其中该字典的键是‘momentum_buffer’,值为相应的动量缓存。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值