浏览代码

add infer code

often 5 月之前
父节点
当前提交
d0243a8541
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

+ 1 - 0
recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

@@ -25,6 +25,7 @@ import sys
 import paddle.distributed.fleet as fleet
 import paddle.distributed.fleet.base.role_maker as role_maker
 import paddle
+import threading
 
 import warnings
 import logging