dssm train

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

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