保存paddle模型时infer_slots_data插入位置导致预估出core

gr8qqesn 于 5个月前发布在其他

关注(0)|答案(3)|浏览(45)

请提出你的问题 Please ask your question

将infer_slots_data 放到sequence_pool 之前，预估出core，放到sequence_pool 之后，则不出core，求问原因

for data in slots:
emb = fluid.layers.embedding(input=data, size=[self._dict_dim, self._emb_dim], is_sparse=True, is_distributed=True, param_attr=fluid.ParamAttr(name="embedding"))
self.infer_slots_data.append(emb)
bow = fluid.layers.sequence_pool(input=emb, pool_type='sum')
#self.infer_slots_data.append(emb)

core信息：
#0 0x00007f1fba52422b in paddle::operators::ConcatOp::GetExpectedKernelType(paddle::framework::ExecutionContext const&) const ()

Paddle

来源：https://github.com/PaddlePaddle/Paddle/issues/51143

3条答案

按热度按时间

ih99xse11#

infer_slots_data是什么呢，可以贴一下代码吗

赞(0）回复(0）举报 5个月前

yrwegjxp2#

是save模型的入参feeded_var_names
fluid.io.save_inference_model(
dirname=model_name,
feeded_var_names=feeded_var_names,
target_vars=target_vars,
executor=executor,
#main_program=program.clone(),
main_program=prog,
params_filename="params")

赞(0）回复(0）举报 5个月前

cgh8pdjw3#

最终是修改 config.switch_ir_optim(False) ，关闭预测引擎优化解决问题，paddle version = 1.8.5.

赞(0）回复(0）举报 5个月前