Toybrick

标题: 算法分类是否支持多分类 [打印本页]

作者: czcl 时间: 2023-2-1 10:25
标题: 算法分类是否支持多分类
问题：
使用pytorch 模型经过转换，变成rknn的模型(自己训练的多分类模型)
使用pytorch 验证模型没问题，但是使用rknn的模型验证，结果总是不对

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

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

作者: jefferyzhang 时间: 2023-2-1 11:55
本帖最后由 jefferyzhang 于 2023-2-1 12:07 编辑

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

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