丁云鹏 4 miesięcy temu
rodzic
commit
c6f36790a7

+ 2 - 1
recommend-model-produce/src/main/python/tools/static_ps_infer_v3.py

@@ -296,12 +296,13 @@ class Main(object):
         test_dataloader = self.reader
 
         for batch_id, batch_data in enumerate(test_dataloader()):
+
             fetch_batch_var = exe.run(
                 program=paddle.static.default_main_program(),
                 feed=dict(zip(input_data_names, batch_data)),
                 fetch_list=fetch_vars)
 
-            logger.info("fetch_batch_var : {}".format(fetch_batch_var))
+            logger.info("batch_data:{},fetch_batch_var : {}".format(batch_data, fetch_batch_var))
 
 
     def heter_train_loop(self, epoch):