Merge pull request awslabs#3 from kevinthesun/FixTFConfig

Fix tf config file
mseth10 · Nov 30, 2017 · 4786014 · 4786014
2 parents 20051a2 + eb029a3
commit 4786014
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/task_config_template.cfg b/task_config_template.cfg
@@ -44,19 +44,19 @@ num_gpus = 8
 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=cifar10 --num_gpus=2 --batch_size=32 --model=resnet50 --variable_update=parameter_server --print_training_accuracy=True
-num_gpus = 2
+command_to_execute = python tensorflow_benchmark/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_name=cifar10 --num_gpus=1 --batch_size=32 --model=resnet50 --variable_update=parameter_server --print_training_accuracy=True
+num_gpus = 1
 
 [tensorflow_resnet50_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=cifar10 --num_gpus=8 --batch_size=32 --model=resnet50 --variable_update=parameter_server --print_training_accuracy=True
-num_gpus = 8
+command_to_execute = python tensorflow_benchmark/tf_cnn_benchmarks/tf_cnn_benchmarks.py --data_name=cifar10 --num_gpus=4 --batch_size=32 --model=resnet50 --variable_update=parameter_server --print_training_accuracy=True
+num_gpus = 4
 
 [tensorflow_resnet50_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=cifar10 --num_gpus=16 --batch_size=32 --model=resnet50 --variable_update=parameter_server --print_training_accuracy=True
-num_gpus = 16
+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