dssm train

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

@@ -76,7 +76,7 @@ class InferenceFetchHandler(FetchHandler):
         for key in fetch_vars:
             # 转换数据类型
             if type(fetch_vars[key]) is np.ndarray:
-                result = fetch_vars[key][0]
+                result = fetch_vars[key][0].tolist()
             else:
                 result = fetch_vars[key]
             result_dict[key] = result
@@ -87,7 +87,7 @@ class InferenceFetchHandler(FetchHandler):
         while True:
             try:
                 result_dict = self.result_queue.get(timeout=1)  # 非阻塞获取
-                logger.info("write vector {}".format(result_dict))
+                logger.info("write vector {} {}".format(result_dict, len(batch)))
                 batch.append(result_dict)
                 if len(batch) >= self.batch_size:
                     logger.info("write vector")