使用SymbolTree.get_network处理conv2d算子时报错NameError:name "Cell" is not defined
收藏回复举报
使用SymbolTree.get_network处理conv2d算子时报错NameError:name "Cell" is not defined
发表于2023-07-29 16:12:14
0 查看

1 系统环境

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

MindSpore版本: mindspore=2.0

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

Python版本: Python=3.7

操作系统平台: 不限

2 报错信息

2.1 问题描述

在执行stree.get_network()语句时报错,报错信息如下图所示:

2.2 报错信息

NameError:name "Cell" is not defined

2.3 脚本代码(代码格式,可上传附件)

import mindspore.nn as nn

old_model=nn.Conv2d(3,64,1,1)
stree = SymbolTree.create(old_model)
copy_model = stree.get_network()

3 根因分析

******此处由用户补充详细的定位过程******

参考官网查看SymbolTree.create的用法

4 解决方案

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

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

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

00.PNG

我要发帖子