dssm train

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

@@ -77,9 +77,9 @@ class InferenceFetchHandler(FetchHandler):
         for key in fetch_vars:
             # 转换数据类型
             if type(fetch_vars[key]) is np.ndarray:
-                result = res_dict[key][0]
+                result = fetch_vars[key][0]
             else:
-                result = res_dict[key]
+                result = fetch_vars[key]
             result_dict[key] = result
         self.result_queue.put(result_dict)  # 将结果放入队列