RK1808计算棒是否支持下面的功能？

[复制链接] · 发表于 2019-10-25 18:22:28

购买该计算棒，但是资料很少，所以想请教一下的功能是否支持1.是否支持网络Pruning，就是可以手动将网络的相关权重设置为0
2.Custom node支持到什么程度，有没有详细的举例呢
3.提供的工具支持retraining吗？可以重新训练模型

只看该作者 · 发表于 2019-10-27 01:02:03

资料并不少，网站上wiki、教程、直播全都是，1808和3399pro所有流程都是一样的。
1. 你可以把权重设0后再转rknn
2. 支持到所有都可以自己写，可以通过openVX调用NPU，也可以通过c语言调用cpu运算。详细例子在附带的文档和demo里都有
3. 可以自己转换

只看该作者 · 发表于 2019-10-28 11:45:02

jefferyzhang 发表于 2019-10-27 01:02
资料并不少，网站上wiki、教程、直播全都是，1808和3399pro所有流程都是一样的。
1. 你可以把权重设0后再转 ...

感谢你的解答，可能是自己没有将资料找全
请教一下：
1.Pruning 想要的效果，能否在给定的模型文件中通过咱们的工具在工具转换中进行修改权重为0，不需要要在模型训练中操作，不知道现在的工具是否可以，
2.这个附带的文档我再寻找一下。这里想询问的Custom node是指在不重写模型，在转换为rknn时能够使用工具自行拓展一些模型部分
3.这个可以理解为带的工具并不带重新训练模型的部分，若想对这个预训练的模型进行重新的训练，必须自己写转换并重新训练对吧

只看该作者 · 发表于 2019-10-28 12:01:35

jefferyzhang 发表于 2019-10-27 01:02
资料并不少，网站上wiki、教程、直播全都是，1808和3399pro所有流程都是一样的。
1. 你可以把权重设0后再转 ...

顺带再问一下，先关的工具支持GPU的转换吗

只看该作者 · 发表于 2019-10-28 14:08:51

mailonghua 发表于 2019-10-28 12:01
顺带再问一下，先关的工具支持GPU的转换吗

不支持。
RKNN所指的均是NPU端的。

只看该作者 · 发表于 2019-10-28 14:14:55

mailonghua 发表于 2019-10-28 11:45
感谢你的解答，可能是自己没有将资料找全
请教一下：
1.Pruning 想要的效果，能否在给定的模型文件中通过 ...

1. 目前没有该功能，暂时无法实现。
2. 是的，自定义OP是用来解决手动增加不支持的OP算子。最不济就是全部CPU处理，不存在不支持的可能。
3. rknn只是从源模型到目标NPU的一个映射转换而已，rknn并不是新的框架语言。并且我们的NPU是不支持训练的。

只看该作者 · 发表于 2019-10-28 18:17:43

jefferyzhang 发表于 2019-10-28 14:14
1. 目前没有该功能，暂时无法实现。
2. 是的，自定义OP是用来解决手动增加不支持的OP算子。最不济就是全 ...

感谢你的解答，非常详细，自己再琢磨琢磨