如何使用MindSpore替换PyTorch的torch.nn.init
收藏回复举报
如何使用MindSpore替换PyTorch的torch.nn.init
发表于2024-02-23 08:42:05
0 查看

系统环境

硬件环境(Ascend/GPU/CPU): Ascend/GPU/CPU

MindSpore版本: mindspore=2.0.0

执行模式(PyNative/ Graph):不限

Python版本: Python=3.7

操作系统平台: 不限

报错信息

2.1问题描述

做pytorch迁移到MindSpore的过程中需要更改调用torch.nn.init相关的部分,MindSpore该如何替换

根因分析

******此处由用户填写******

解决方案

******此处由用户填写******

包含文字方案和最终脚本代码

请将正确的脚本打包并上传附件

我要发帖子