丁云鹏 4 ماه پیش
والد
کامیت
1acb514e85
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

@@ -79,7 +79,7 @@ class InferenceFetchHandler(FetchHandler):
             if type(fetch_vars[key]) is np.ndarray:
                 result = res_dict[key][0]
             else:
-                result = var_value
+                result = res_dict[key]
             result_dict[key] = result
         self.result_queue.put(result_dict)  # 将结果放入队列