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.

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.
c0a380d2 · Qianli Scott Zhu · GitHub · b9ca525f · c0a380d2 · c0a380d2
Unverified Commit c0a380d2 authored May 11, 2018 by Qianli Scott Zhu Committed by GitHub May 11, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/transformer/transformer_main.py official/transformer/transformer_main.py +1 -1

official/wide_deep/wide_deep.py official/wide_deep/wide_deep.py +1 -1

No files found.
--- a/official/transformer/transformer_main.py
+++ b/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",

--- a/official/wide_deep/wide_deep.py
+++ b/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, '')