dnn test

recommend-model-produce/src/main/python/models/dnn/config.yaml

@@ -21,7 +21,7 @@ runner:
   train_reader_path: "criteo_reader" # importlib format
   use_gpu: False
   use_auc: False
-  train_batch_size: 2
+  train_batch_size: 100
   epochs: 1
   print_interval: 2
   model_save_path: "output_model_dnn"

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

@@ -260,6 +260,7 @@ def dataset_train(epoch_id, dataset, fetch_vars, exe, config):
     fetch_info = [
         "Epoch {} Var {}".format(epoch_id, var_name) for var_name in fetch_vars
     ]
+    logger.info("var_name: {}".format(var_name)) for var_name in fetch_vars
     fetch_vars = [var for _, var in fetch_vars.items()]
     print_interval = config.get("runner.print_interval")
     exe.train_from_dataset(