Skip to content

M
models

PaddlePaddle / models
大约 2 年 前同步成功

通知 232
Star 6828
Fork 2962
M
models
提交 3a784aa0 编写于 作者: D dzhwinter 提交者： GitHub
浏览文件
操作

Merge pull request #171 from dzhwinter/fix_ltr 

"rename infer network name"
上级 63fc48b0 883006a0
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
ltr/ranknet.py
浏览文件 @ 3a784aa0
...@@ -102,7 +102,7 @@ def ranknet_infer(pass_id): ...@@ -102,7 +102,7 @@ def ranknet_infer(pass_id):
feature_dim = 46 feature_dim = 46
# we just need half_ranknet to predict a rank score, which can be used in sort documents # we just need half_ranknet to predict a rank score, which can be used in sort documents
output = half_ranknet("left", feature_dim) output = half_ranknet("infer", feature_dim)
parameters = paddle.parameters.Parameters.from_tar( parameters = paddle.parameters.Parameters.from_tar(
gzip.open("ranknet_params_%d.tar.gz" % (pass_id - 1))) gzip.open("ranknet_params_%d.tar.gz" % (pass_id - 1)))
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或