rknn-toolkit支持pytorch1.4.0下efficientDetD1的模型为rknn模型吗

我们做芯片角度只考虑op是不是支持，不用问什么什么模型支不支持，模型每年增长上千个，我们怎么可能一个个试过去，转下看看有没有报错就知道了。
目前rknntoolkit 1.3.2只支持pytorch 1.2.0，其他版本不支持。
还有efficientnet不是google发明的网络么，为啥不用tf要用pytorch转

senna_kim 发表于 2020-8-11 15:41
你好，请问一下，这种情况 是不是需要自己定义Op?

log上看 是FusedBatchNormV3 这个不支持。
你可以考虑修改模型比较快。

senna_kim 发表于 2020-8-11 16:37
你的意思是移除这一层网络吗？还是。。。
请再多给点细节，谢谢！

改普通BN层试试，改完要重新训练

senna_kim 发表于 2020-8-11 18:17
你们现在硬件优化的Op都支持哪些？有文档可以看一下吗？

没有文档，硬件op和软件op不是一个概念，每个框架兼容程度也不一样。我们尽可能的全兼容，但新op出的速度远快于我们支持速度。不支持的可以自己通过自定义op来支持，或者等我们支持。