Toybrick

mxnet 全连接层 flatten=True时,无法转化模型

jiaruoran

新手上路

积分
11
楼主
发表于 2021-6-11 16:18:08    查看: 6842|回复: 0 | [复制链接]    打印 | 只看该作者
我在进行mxnet模型和rknn转化时发现一个问题,当全连接层  nn.Dense(voc_size, flatten=True), flatten为True时即常规的全连接操作可以正常转化计算,而当flatten=false即只对最后一维进行全连接操作时,mxnet模型无法转化为rknn模型,请问是否有办法解决这个问题
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表