MindSpore模型的Parameter在图模式下如何手动更新

目前实现了类似MoCo的memory bank的结构,使用Parameter构造,在模型运行过程中需要更新,代码如下

keys = ops.stop_gradient(keys)
labels = ops.stop_gradient(labels)
flag = ops.stop_gradient(flag)
batch_size = keys.shape[0]
ptr = int(self.ptr)
self.queue[ptr:ptr + batch_size, :] = keys
self.queue_l[ptr:ptr + batch_size] = labels
ptr = (ptr + batch_size) % self.K
self.ptr[0] = ptr

Parameter定义方式如下

self.queue = Parameter(Tensor(randn(self.K, s), mstype.float32))
self.queue = Parameter(normalize(self.queue), "queue", False, False, False)
self.queue_l = Parameter(Tensor(randint(0, 2, (self.K, l)), mstype.float32), "queue_l", False, False, False)
self.ptr = Parameter(zeros(1, mstype.int64), "ptr", False, False, False)

在PYNATIVE模式下正常运行,在GRAPH模式下运行报错 ptr:ptr + bach_size invalid syntax

如何修改以运行在GRAPH模式下

****************************************************解答*****************************************************

本地试了一下,ptr = int(self.ptr)会报错,constrcut里不支持对Parameter(非常量Tensor)类型转换。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值