Toybrick

请问多久会支持pytorch框架?据说onnx要散伙了

kitedream

中级会员

积分
284
发表于 2019-3-3 10:55:59    查看: 23706|回复: 13 | [复制链接]    打印 | 显示全部楼层
现在pytorch 越来越流行了,很多新论文是用pytorch 实现的
回复

使用道具 举报

jefferyzhang

版主

积分
13098
发表于 2019-3-3 18:38:52 | 显示全部楼层
keras和pytorch我都已经向有关部门提过需求了,一起等吧,我相信不久的将来就会支持
毕竟目前最流行的就是这俩了
回复

使用道具 举报

leok

版主

积分
894
发表于 2019-3-4 02:13:05 | 显示全部楼层
目前可以这样玩,先转成onnx,再转成rknn。
回复

使用道具 举报

jefferyzhang

版主

积分
13098
发表于 2019-3-7 09:24:37 | 显示全部楼层
本帖最后由 jefferyzhang 于 2019-4-25 09:34 编辑

已经在支持中了,未来将会发布
回复

使用道具 举报

kitedream

中级会员

积分
284
 楼主| 发表于 2019-3-7 14:14:33 | 显示全部楼层
jefferyzhang 发表于 2019-3-7 09:24
确切消息,已经在支持中了,下一个版本应该会发布

激动
回复

使用道具 举报

chuyee

中级会员

积分
352
发表于 2019-3-11 03:29:46 | 显示全部楼层
When will be the next release? Can't wait for the pytorch support!
回复

使用道具 举报

granis

注册会员

积分
73
发表于 2019-3-12 18:36:32 | 显示全部楼层
jefferyzhang 发表于 2019-3-7 09:24
确切消息,已经在支持中了,下一个版本应该会发布

pytorch过来onnx转换的时候反卷积和Clip(0,6)都有问题。
能给支持么?
回复

使用道具 举报

jefferyzhang

版主

积分
13098
发表于 2019-3-14 09:02:10 | 显示全部楼层
granis 发表于 2019-3-12 18:36
pytorch过来onnx转换的时候反卷积和Clip(0,6)都有问题。
能给支持么?

反卷积支持,
Clip暂时不支持
回复

使用道具 举报

granis

注册会员

积分
73
发表于 2019-3-14 09:14:49 | 显示全部楼层
jefferyzhang 发表于 2019-3-14 09:02
反卷积支持,
Clip暂时不支持

帮我看看这个,在反卷积的时候出错。Clip是MobileNetV2中用到的,能否提供支持,实现起来也是非常简单。
W verbose file path is invalid, debug info will not dump to file.
I Current ONNX Model use ir_version 3 opset_version 9
D import clients finished
I build output layer output_473
I Try match Conv op#154
I Match [['473']] [['Conv_1']] to [['convolution_1']]
I Try match Relu op#153
I Match [['472']] [['Relu_1']] to [['relu_1']]
I Try match BatchNormalization op#152
I Match [['471']] [['BatchNormalization']] to [['batchnormalize_1']]
I Try match ConvTranspose op#151
I Match [['470']] [['ConvTranspose_1']] to [['deconvolution_1']]
E Catch exception when loading onnx model: ./model.onnx!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 321, in rknn.api.rknn_base.RKNNBase.load_onnx
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 555, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 355, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_build_acu_layer
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 322, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_acu_param_assign
T   File "rknn/base/rknnlib/converter/convert_onnx.py", line 302, in rknn.base.rknnlib.converter.convert_onnx.convert_onnx._onnx_parse_param
T   File "rknn/base/rknnlib/converter/paragraph.py", line 21, in rknn.base.rknnlib.converter.paragraph.lambda4
T TypeError: 'int' object is not iterable
回复

使用道具 举报

jefferyzhang

版主

积分
13098
发表于 2019-3-14 09:25:52 | 显示全部楼层
granis 发表于 2019-3-14 09:14
帮我看看这个,在反卷积的时候出错。Clip是MobileNetV2中用到的,能否提供支持,实现起来也是非常简单。
...

这个log看是匹配反卷积层时候出现问题,并不是op不支持。
先测试下你的模型转成onnx后是否可以正常推理,再转rknn

不支持Clip op我们报告给相关部门去处理了。
回复

使用道具 举报

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

本版积分规则

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


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