专栏算法工具链自己定义的lenet模型不支持

自己定义的lenet模型不支持

已解决
wangning2020-12-12
99
2

您好,我们自己定义的lenet模型在runtime_sim下的推理运行时报错:

请问这个问题该如何解决呢?

注释:这个问题我们在后处理源码文件中做了如下修改:

算法工具链
评论2
0/1000
  • 孙宏伟
    Lv.1

    由于自己训练的lent模型,在最后一层没有添加softmax,因此导致出现问题,

    具体修改方法,参考源码:3_common/src/post_process/classification_post_process.cc

    2020-12-12
    2
    0
  • 孙宏伟
    Lv.1

    已解决

    2020-12-12
    0
    0