From 96930218579d4da8950638bca745924323a28dd5 Mon Sep 17 00:00:00 2001 From: Wang Date: Fri, 1 Dec 2017 11:40:53 -0800 Subject: [PATCH] Add tensorflow resnet152 imagenet --- task_config_template.cfg | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/task_config_template.cfg b/task_config_template.cfg index 90a0c79..73d42b9 100644 --- a/task_config_template.cfg +++ b/task_config_template.cfg @@ -60,3 +60,24 @@ metrics = ['Images per sec', 'Top 1 accuracy'] compute_method = ['average', 'last'] command_to_execute = python tensorflow_benchmark/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_name=cifar10 --num_gpus=8 --batch_size=32 --model=resnet50 --variable_update=parameter_server --print_training_accuracy=True num_gpus = 8 + +[tensorflow_resnet152_p3_2xlg] +patterns = ['images/sec: (\d+\.\d+)', 'top_1_accuracy: (\d+\.\d+)'] +metrics = ['Images per sec', 'Top 1 accuracy'] +compute_method = ['average', 'last'] +command_to_execute = python tensorflow_benchmark/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_name=imagenet --num_gpus=1 --batch_size=32 --model=resnet152 --variable_update=parameter_server --print_training_accuracy=True +num_gpus = 1 + +[tensorflow_resnet152_p3_8xlg] +patterns = ['images/sec: (\d+\.\d+)', 'top_1_accuracy: (\d+\.\d+)'] +metrics = ['Images per sec', 'Top 1 accuracy'] +compute_method = ['average', 'last'] +command_to_execute = python tensorflow_benchmark/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_name=imagenet --num_gpus=4 --batch_size=32 --model=resnet152 --variable_update=parameter_server --print_training_accuracy=True +num_gpus = 4 + +[tensorflow_resnet152_p3_16xlg] +patterns = ['images/sec: (\d+\.\d+)', 'top_1_accuracy: (\d+\.\d+)'] +metrics = ['Images per sec', 'Top 1 accuracy'] +compute_method = ['average', 'last'] +command_to_execute = python tensorflow_benchmark/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_name=imagenet --num_gpus=8 --batch_size=32 --model=resnet152 --variable_update=parameter_server --print_training_accuracy=True +num_gpus = 8