全部回复
LV.1
#2
w3jc
回复
需要补充的代码可以参考文件里run_ori_pb()函数的实现,至于需要的张量名称可以把pb模型转换为pbtxt查看,或者用netron(
https://lutzroeder.github.io/netron/
)之类的工具。
transform_mlu.py和
transform_cpu.py补完的代码基本一致,就mlu里面多一句保存离线模型config.mlu_options.save_offline_model = True。
2020-07-01 12:04:58
1
|
回复
优质回复
LV.2
#1
希望
回复
具体内容的编写需参考模型文件对参数的要求,具体在
models/int_pb_models/
下,可使用
Netron
软件进行可视化查看
2020-07-01 11:54:12
0
|
回复
优质回复
请
登录
后评论