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

qlzh727 · web-flow · commit c0a380d21f39 · 2018-05-11T14:48:19.000-07:00
* Update the wide_deep code for latest benchmark config.

* Also update the transformer benchmark code.
diff --git 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",
diff --git a/official/wide_deep/wide_deep.py b/official/wide_deep/wide_deep.py
@@ -245,7 +245,7 @@ def eval_input_fn():
       '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, '')

Original file line number	Diff line number	Diff line change
`@@ -433,7 +433,7 @@ def run_transformer(flags_obj):`
`433`	`433`	`tensors_to_log=TENSORS_TO_LOG, # used for logging hooks`
`434`	`434`	`batch_size=params.batch_size # for ExamplesPerSecondHook`
`435`	`435`	`)`
`436`		`- benchmark_logger = logger.config_benchmark_logger(flags_obj.benchmark_log_dir)`
	`436`	`+ benchmark_logger = logger.config_benchmark_logger(flags_obj)`
`437`	`437`	`benchmark_logger.log_run_info(`
`438`	`438`	`model_name="transformer",`
`439`	`439`	`dataset_name="wmt_translate_ende",`
Original file line number	Diff line number	Diff line change
`@@ -245,7 +245,7 @@ def eval_input_fn():`
`245`	`245`	`'model_type': flags_obj.model_type,`
`246`	`246`	`}`
`247`	`247`
`248`		`- benchmark_logger = logger.config_benchmark_logger(flags_obj.benchmark_log_dir)`
	`248`	`+ benchmark_logger = logger.config_benchmark_logger(flags_obj)`
`249`	`249`	`benchmark_logger.log_run_info('wide_deep', 'Census Income', run_params)`
`250`	`250`
`251`	`251`	`loss_prefix = LOSS_PREFIX.get(flags_obj.model_type, '')`