关于Pytorch中Conv2d转rknn的问题

只看该作者 · 发表于 2020-4-14 08:16:38

LudvigCL 发表于 2020-4-13 16:29
请问量化时如果指定为npy文件，那其中的数组是必须按照NHWC格式进行保存吗？
也即是说如果我在jit.trace ...

回复如下：

必须按照NHWC的格式。
ps.如果使用npy文件量化并且初始化设置如下
rknn=RKNN(verbose=True)
就会有一个Warning警告
W xxx.npy need NHWC format!

只看该作者 · 发表于 2020-4-14 09:52:46

jefferyzhang 发表于 2020-4-13 22:37
你是在Init runtime environment时候出错，请发下你toolkit、api、drv三者的版本号分别是多少 ...

Init runtime environment是在pc上进行仿真时候出错的，使用的toolkit是1.3.1b7版本

只看该作者 · 发表于 2020-4-14 09:53:15

jefferyzhang 发表于 2020-4-14 08:16
回复如下：

必须按照NHWC的格式。

好的多谢指点

只看该作者 · 发表于 2020-4-14 09:54:33

LudvigCL 发表于 2020-4-14 09:52
Init runtime environment是在pc上进行仿真时候出错的，使用的toolkit是1.3.1b7版本

beta7在仿真时候是会出问题的，这个只是beta版本。可以试试用1.3.0版来加载这个模型，
不然你只能等正式版出来看看了。

只看该作者 · 发表于 2020-4-14 10:07:50

jefferyzhang 发表于 2020-4-14 09:54
beta7在仿真时候是会出问题的，这个只是beta版本。可以试试用1.3.0版来加载这个模型，
不然你只能等正式 ...

好的，抱歉劳烦您了

只看该作者 · 发表于 2020-4-17 14:50:23

jefferyzhang 发表于 2020-4-14 09:54
beta7在仿真时候是会出问题的，这个只是beta版本。可以试试用1.3.0版来加载这个模型，
不然你只能等正式 ...

BETA版本在哪里可以找到呢？

只看该作者 · 发表于 2020-4-17 15:03:44

luxiaofei 发表于 2020-4-17 14:50
BETA版本在哪里可以找到呢？