18 lines
558 B
Python
18 lines
558 B
Python
# paddle模型转nb格式，推荐使用paddlelite2.10
|
||
from paddlelite.lite import Opt
|
||
|
||
if __name__ == '__main__':
|
||
# 1. 创建opt实例
|
||
opt = Opt()
|
||
# 2. 指定输入模型地址
|
||
opt.set_model_dir("./model")
|
||
# 3. 指定转化类型： arm、x86、opencl、npu
|
||
# 一般认为arm为cpu，opencl为gpu
|
||
opt.set_valid_places("arm")
|
||
# 4. 指定模型转化类型： naive_buffer、protobuf
|
||
opt.set_model_type("naive_buffer")
|
||
# 4. 输出模型地址
|
||
opt.set_optimize_out("model")
|
||
# 5. 执行模型优化
|
||
opt.run()
|