Toybrick

算法分类是否支持多分类

czcl

中级会员

积分
365
发表于 2023-2-1 10:25:05    查看: 1180|回复: 1 | [复制链接]    打印 | 显示全部楼层
问题:
使用pytorch 模型经过转换,变成rknn的模型(自己训练的多分类模型)
使用pytorch 验证模型没问题,但是使用rknn的模型验证,结果总是不对

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


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


回复

使用道具 举报

jefferyzhang

版主

积分
12851
发表于 2023-2-1 11:55:33 | 显示全部楼层
本帖最后由 jefferyzhang 于 2023-2-1 12:07 编辑

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

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表