专栏算法工具链hb_mapper makertbin 进行模型转换时 遇到输入和模型通道不匹配的错误

hb_mapper makertbin 进行模型转换时 遇到输入和模型通道不匹配的错误

已解决
素质诗人2020-10-22
146
2

在尝试将自己在CIFAR10数据集上训练的DenseNet40转换成量化模型的过程中，使用命令hb_mapper makertbin，报以下错误：

请问是NCHW和NHWC格式冲突的问题吗？但在配置文件中两种参数都使用了，都报同样的错误

算法工具链
评论1
0/1000
  • anming
    Lv.1

    讲训练数据输入改为3x32x32， 重行训练。然后参照 mobilenet_config.yaml 编译

    2020-10-22
    0
    1
    • 素质诗人回复anming:

      所以所有模型只支持NCHW的格式嘛

      2020-10-22
      1