Skip to content

Commit

Permalink
Merge pull request awslabs#4 from kevinthesun/AddTFResnet152
Browse files Browse the repository at this point in the history
Add tensorflow resnet152 imagenet
  • Loading branch information
jiajiechen authored Dec 1, 2017
2 parents 4786014 + 9693021 commit 71db1c8
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions task_config_template.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 71db1c8

Please sign in to comment.