算法分类是否支持多分类

[复制链接] · 发表于 2023-2-1 10:25:05

问题：
使用pytorch 模型经过转换，变成rknn的模型(自己训练的多分类模型)
使用pytorch 验证模型没问题，但是使用rknn的模型验证，结果总是不对

如果有多分类的demo请提供一下，(示例中的都是单分类的)

瑞新微的npu芯片，是否支持多分类，或者说多分类有什么具体的要求么？

只看该作者 · 发表于 2023-2-1 11:55:33

本帖最后由 jefferyzhang 于 2023-2-1 12:07 编辑

示例都是多分类啊，哪有单分类的例子。
你是指多分类还是多输出？多输出的话先把模型input/output的attr query出来看下对不对