X2Paddle
X2Paddle copied to clipboard
转换pytorch的mobilenetv3到lite格式报错不支持
https://github.com/PaddlePaddle/Paddle-Lite/issues/8908#issue-1210769522
# 进行转换
from x2paddle.convert import pytorch2paddle
pytorch2paddle(torch_model,
save_dir="pd_model_trace_mbv3",
jit_type="trace",
input_examples=[torch.tensor(input_data)]
convert_to_lite=True,
lite_valid_places="arm",
lite_model_type="naive_buffer"
)
用上面的代码转换pytorch的mobilenetv3到lite格式,抛错不支持:
已经到paddlelite那边提交issue说,lite已经支持这些算子,让更新X2paddle版本,更新了X2paddle到1.3.6也还是这个抛错。请问X2paddle目前不支持mobilenetV3的转换吗,请问下这是什么原因,有什么解决方法,感谢
@BgLoveXixi 目前aten::linear
已经支持,另外两个会排期支持下~
之前有看到您在转AlexNet,都是些分类模型,请描述下业务场景吧~
@wjj19950828 好的,能支持了麻烦告知一下,是会在最新版里面支持对吗?
是的,之前转了AlexNet的分类模型,业务场景是想在手机端实现皮肤病的初筛
@BgLoveXixi 麻烦提供一下组网、模型参数以及转换脚本,打包上传到百度云吧
@wjj19950828 相关的文件都传到网盘了: 链接:https://pan.baidu.com/s/1dH4a3myBTqT62EQN-9-NDA?pwd=zlb5 提取码:zlb5
@wjj19950828 我尝试转换mobilenetv2的pytorch模型到lite格式,也还是报错有一个算子不支持,是aten::linear,可是这个算子您不是说已经支持了吗
请问这个怎么解决呢?我该更新x2paddl的源码吗。
还有就是mobilenetv3那里面的几个算子什么大概时候能安排支持下呢,感谢!
@BgLoveXixi aten::linear
在这个PR:https://github.com/PaddlePaddle/X2Paddle/pull/769
还没合入,可以先拉去对应分支手动编译一下
另外两个op正在开发当中
@BgLoveXixi 您好,目前三个op的支持都已经合入,拉取最新的develop分支代码手动安装即可~
另外可以加您一个微信吗,想对您做一下用户访谈,感谢~
@wjj19950828 好的收到谢谢,联系方式发您邮箱了