Update the wide_deep and transformer code for latest benchmark config. (#4246)

* Update the wide_deep code for latest benchmark config.

* Also update the transformer benchmark code.

official/transformer/transformer_main.py

@@ -433,7 +433,7 @@ def run_transformer(flags_obj):
       tensors_to_log=TENSORS_TO_LOG,  # used for logging hooks
       batch_size=params.batch_size  # for ExamplesPerSecondHook
   )
-  benchmark_logger = logger.config_benchmark_logger(flags_obj.benchmark_log_dir)
+  benchmark_logger = logger.config_benchmark_logger(flags_obj)
   benchmark_logger.log_run_info(
       model_name="transformer",
       dataset_name="wmt_translate_ende",

official/wide_deep/wide_deep.py

@@ -245,7 +245,7 @@ def run_wide_deep(flags_obj):
       'model_type': flags_obj.model_type,
   }
 
-  benchmark_logger = logger.config_benchmark_logger(flags_obj.benchmark_log_dir)
+  benchmark_logger = logger.config_benchmark_logger(flags_obj)
   benchmark_logger.log_run_info('wide_deep', 'Census Income', run_params)
 
   loss_prefix = LOSS_PREFIX.get(flags_obj.model_type, '')