Toybrick

标题: mxnet 全连接层 flatten=True时，无法转化模型 [打印本页]

作者: jiaruoran 时间: 2021-6-11 16:18
标题: mxnet 全连接层 flatten=True时，无法转化模型
我在进行mxnet模型和rknn转化时发现一个问题，当全连接层 nn.Dense(voc_size, flatten=True)， flatten为True时即常规的全连接操作可以正常转化计算，而当flatten=false即只对最后一维进行全连接操作时，mxnet模型无法转化为rknn模型，请问是否有办法解决这个问题

欢迎光临 Toybrick (https://t.rock-chips.com/)