丁云鹏 5 hónapja
szülő
commit
2a0efa4dba

+ 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')