提交 a9ba9ca3 编写于 作者: 0 0YuanZhang0 提交者: pkpk

fix readme and predict params (#2916)

上级 b550a402
......@@ -253,7 +253,7 @@ task_name: udc, swda, mrda, atis_intent, atis_slot, dstc2
    ii、 自定义上层网络范式
      如果用户自定义模型属于分类、多分类和序列标注这3种类型其中一个,则只需要在**paddle-nlp/models/dialogue_model_toolkit/dialogue_general_understanding/define_paradigm.py** 内指明**task_name**和相应上层范式函数的对应关系即可,如用户自定义模型属于其他模型,则需要自定义上层范式函数并指明其与**task_name**之间的关系;
      如果用户自定义模型属于分类、多分类和序列标注这3种类型其中一个,则只需要在**models/PaddleNLP/models/dialogue_model_toolkit/dialogue_general_understanding/define_paradigm.py** 内指明**task_name**和相应上层范式函数的对应关系即可,如用户自定义模型属于其他模型,则需要自定义上层范式函数并指明其与**task_name**之间的关系;
    iii、自定义预测封装接口
......
......@@ -97,7 +97,7 @@ def main(args):
is_prediction=True)
predict_pyreader = pred_results.get('pyreader', None)
probs = pred_results.get('probs', None)
feed_target_names = pred_results.get('feed_target_names', None)
feed_target_names = pred_results.get('feed_targets_name', None)
predict_prog = predict_prog.clone(for_test=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册