Skip to content

GitLab

"...GATNE-T/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "ee9887d66a9eb6dc9663a5238e1874b7646acf3a"
Unverified Commit d93d1211 authored by Qianli Scott Zhu's avatar Qianli Scott Zhu Committed by GitHub
Browse files

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

This will help us differentiate between dataset.
parent 12e977e4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/resnet/resnet_run_loop.py
View file @ d93d1211
...@@ -395,6 +395,8 @@ def resnet_main( ...@@ -395,6 +395,8 @@ def resnet_main(
'synthetic_data': flags_obj.use_synthetic_data, 'synthetic_data': flags_obj.use_synthetic_data,
'train_epochs': flags_obj.train_epochs, '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 = logger.get_benchmark_logger()
benchmark_logger.log_run_info('resnet', dataset_name, run_params, benchmark_logger.log_run_info('resnet', dataset_name, run_params,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment