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

jefferyzhang 发表于 2020-8-10 16:55
我们做芯片角度只考虑op是不是支持，不用问什么什么模型支不支持，模型每年增长上千个，我们怎么可能一个个 ...

你好，请问一下，这种情况 是不是需要自己定义Op?

1. 
   
2. Try match FusedBatchNormV3 FeatureExtractor/MobilenetV2/layer_19_2_Conv2d_5_3x3_s2_128/BatchNorm/FusedBatchNormV3 failed, catch exception!
   
3. E Catch exception when loading tensorflow model: ./output_inference_graph_v2/output_inference_graph_v2/frozen_inference_graph.pb!
   
4. E Traceback (most recent call last):
   
5. E   File "rknn\base\RKNNlib\converter\convert_tf.py", line 643, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
   
6. E   File "rknn\base\RKNNlib\RKNNnet.py", line 189, in rknn.base.RKNNlib.RKNNnet.RKNNNet.new_layer
   
7. E   File "rknn\base\RKNNlib\RKNNlog.py", line 327, in rknn.base.RKNNlib.RKNNlog.RKNNLog.e
   
8. E ValueError: Unknow layer "fusedbatchnormv3"
   
9. E During handling of the above exception, another exception occurred:
   
10. E Traceback (most recent call last):
    
11. E   File "rknn\api\rknn_base.py", line 303, in rknn.api.rknn_base.RKNNBase.load_tensorflow
    
12. E   File "rknn\base\RKNNlib\converter\convert_tf.py", line 742, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
    
13. E   File "rknn\base\RKNNlib\RKNNlog.py", line 327, in rknn.base.RKNNlib.RKNNlog.RKNNLog.e
    
14. E ValueError: Try match FusedBatchNormV3 FeatureExtractor/MobilenetV2/layer_19_2_Conv2d_5_3x3_s2_128/BatchNorm/FusedBatchNormV3 failed, catch exception!

复制代码

jefferyzhang 发表于 2020-8-11 16:09
log上看 是FusedBatchNormV3 这个不支持。
你可以考虑修改模型比较快。

你的意思是移除这一层网络吗？还是。。。
请再多给点细节，谢谢！

jefferyzhang 发表于 2020-8-11 17:59
改普通BN层试试，改完要重新训练

你们现在硬件优化的Op都支持哪些？有文档可以看一下吗？

jefferyzhang 发表于 2020-8-11 19:55
没有文档，硬件op和软件op不是一个概念，每个框架兼容程度也不一样。我们尽可能的全兼容，但新op出的速度 ...

好的，谢谢！