Update the dataset name for synthetic data. (#4443)

This will help us differentiate between dataset.

Update the dataset name for synthetic data. (#4443)
This will help us differentiate between dataset.
d93d1211 · Qianli Scott Zhu · GitHub · 12e977e4 · d93d1211
Unverified Commit d93d1211 authored Jun 04, 2018 by Qianli Scott Zhu Committed by GitHub Jun 04, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/resnet/resnet_run_loop.py official/resnet/resnet_run_loop.py +2 -0

No files found.
--- a/official/resnet/resnet_run_loop.py
+++ b/official/resnet/resnet_run_loop.py
@@ -395,6 +395,8 @@ def resnet_main(
      'synthetic_data': flags_obj.use_synthetic_data,
      'train_epochs': flags_obj.train_epochs,
  }
+  if flags_obj.use_synthetic_data:
+    dataset_name = dataset_name + "-synthetic"

  benchmark_logger = logger.get_benchmark_logger()
  benchmark_logger.log_run_info('resnet', dataset_name, run_params,