1 系统环境
硬件环境(Ascend/GPU/CPU): GPU/CPU
MindSpore版本: mindspore=2.2.0
执行模式(PyNative/ Graph):不限
Python版本: Python=3.8
操作系统平台: 不限
2 报错信息
2.1问题描述
将pytorch的代码转换为mindspore,其中在embedding层遇到如下代码:
其中input_data为int64类型,torch的embedding能够处理,使用mindspore将代码修改为:
发现定义出来的net的dtype仍为float32,导致了类型错误。
2.2报错信息
3 根因分析
******此处由用户填写******
4 解决方案
******此处由用户填写******
包含文字方案和最终脚本代码
请将正确的脚本打包并上传附件
1 系统环境
硬件环境(Ascend/GPU/CPU): GPU/CPU
MindSpore版本: mindspore=2.2.0
执行模式(PyNative/ Graph):不限
Python版本: Python=3.8
操作系统平台: 不限
2 报错信息
2.1问题描述
将pytorch的代码转换为mindspore,其中在embedding层遇到如下代码:
其中input_data为int64类型,torch的embedding能够处理,使用mindspore将代码修改为:
发现定义出来的net的dtype仍为float32,导致了类型错误。
2.2报错信息
3 根因分析
******此处由用户填写******
4 解决方案
******此处由用户填写******
包含文字方案和最终脚本代码
请将正确的脚本打包并上传附件