pyreader 读入数据,并未使用dataset,如何发布服务
Created by: iceriver97
我该如何将自定义的 Variable:test 绑定到模型中?我看示例是采用的dataset.set_use_var(),但我并未使用dataset,而是使用的pyreader 读入的数据;下面代码不做绑定,直接这样保存模型是不是错误的做法?
test = fluid.layers.data(name="test", shape=[1], dtype="int64", lod_level=1)
prediction = predict_wrapper_serving(reader, exe, test_prog, test_pyreader, graph_vars,'final', 'final')
serving_io.save_model("RE_model","RE_client_conf", {"text": test}, {"prediction": prediction},fluid.default_main_program())
我如果将自定义的test变量传入网络,应该怎样修改呢?网络之前是使用pyreader读入数据的;代码如下:
(src_ids, sent_ids, pos_ids, task_ids, input_mask, labels, seq_lens,
example_index, tok_to_orig_start_index,
tok_to_orig_end_index) = fluid.layers.read_file(pyreader)
@wangjiawei04 使用 pyreader 异步读取数据的模型是否不能发布serving?有没有类似的例子?