瀏覽代碼

dssm train

丁云鹏 5 月之前
父節點
當前提交
2a0efa4dba
共有 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

@@ -91,6 +91,7 @@ class InferenceFetchHandler(FetchHandler):
                 result_dict = self.result_queue.get(timeout=1)  # 非阻塞获取
                 batch.append(result_dict)
                 if len(batch) >= self.batch_size:
+                    logger.info("write vector")
                     with open(self.output_file, 'a') as f:
                         for result in batch:
                             f.write(json.dumps(result) + '\n')