Open czp97 opened 2 weeks ago
.nb 文件的载入方式是不是有问题呢,是不是需要PaddleLite 的 API 进行加载和推理
.nb 文件的载入方式是不是有问题呢,是不是需要PaddleLite 的 API 进行加载和推理
根据PaddleLite给的 PaddleLite-generic-demo.tar.gz 示例,可以使用Paddle载入nb模型并进行推理 具体代码如下:
import paddle
import paddle.fluid as fluid
if len(MODEL_FILE) == 0 and len(PARAMS_FILE) == 0:
[program, feed_target_names,
fetch_targets] = fluid.io.load_inference_model(model_dir, exe)
else:
[program, feed_target_names,
fetch_targets] = fluid.io.load_inference_model(
model_dir,
exe,
model_filename=MODEL_FILE,
params_filename=PARAMS_FILE)
其中,model_dir
是.nb文件所在路径,model_filename
是paddllite生成的model
和params
文件路径
请提出你的问题 Please ask your question
运行环境为: Kylinv10 OS Paddle 2.6.0 PaddleSlim 2.6.1 FT2000+ CPU 昆仑芯R200 XPU
原始模型为Pytorch导出的Resnet50转Paddle模型 PTQ代码如下:
PaddleLite opt转换代码如下:
导出paddle-lite模型int8.nb、model、param
使用Paddle载入int8.nb模型时报错:
使用Paddle载入model、param时报错如下:
量化模型文件提取码:bvxo