栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch代码迁移到mindspore时,API的替换问题

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pytorch代码迁移到mindspore时,API的替换问题

问题描述:

问题

pytorch在训练时采用

ptimizer.zero_grad(),

loss.backward(),

optimizer.step()

这三步流程。

我发现mindspore在这里的设计跟pytorch不一样。

想问一下有没有尽可能改动少的一点的修改方案?

解答:

MindSpore的求导类似pytorch的 torch.autograd.grad(链接torch.autograd.grad — PyTorch 1.12 documentation),将梯度与Tensor剥离,grad操作直接返回梯度,因此不需要 optimizer.zero_grad(), loss.backward() 这两个步骤。

由于 Tensor与梯度剥离,无法直接通过 optimizer.step() 进行参数更新,因此,在MindSpore中更新参数的方式为 optimizer(grads),可以参考 https://www.mindspore.cn/docs/api/zh-CN/r1.5/_modules/mindspore/nn/wrap/cell_wrapper.html#TrainOneStepCell 中construct里的写法。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1002326.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号