dssm train

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

@@ -12,7 +12,6 @@ import paddle.inference as paddle_infer
 
 
 logging.basicConfig(
-    filename='vec.log', filemode='w',
     format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
 logger = logging.getLogger(__name__)
 
@@ -70,8 +69,8 @@ def main():
         predictor.run()
         # 获取输出
         output_data = output_handle.copy_to_cpu() # numpy.ndarray类型
-        print("input k:{} v:{}".format(k, v))
-        print("Output {}".format(output_data))
+        logger.info("input k:{} v:{}".format(k, v))
+        logger.info("Output {}".format(output_data))
 
 if __name__ == "__main__":
     main()