浏览代码

add infer code

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

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

@@ -185,7 +185,7 @@ class Main(object):
         self.metrics = self.model.net(self.inference_feed_var,is_infer=True)
         self.inference_target_var = self.model.inference_target_var
         logger.info("cpu_num: {}".format(os.getenv("CPU_NUM")))
-        self.model.create_optimizer(get_strategy(self.config))
+        self.model.create_optimizer(get_strategy(self.config),is_infer=True)
 
     def run_server(self):
         logger.info("Run Server Begin")