Skip to content

Commit 215179b

Browse files
authored
Add Horovod benchmark (#157)
* Add horovod benchmark
1 parent ce47c76 commit 215179b

File tree

11 files changed

+1475
-265
lines changed

11 files changed

+1475
-265
lines changed

.gitignore

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,7 @@ dist
33
**/*.egg-info
44
.DS_Store
55
.idea/
6-
*.iml
6+
*.iml
7+
**/.ipynb_checkpoints
8+
**/.python-version
9+
.tox

benchmarks/execute_tensorflow_training.py

Lines changed: 0 additions & 264 deletions
This file was deleted.

benchmarks/horovod-resnet/bench.sh

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
#!/usr/bin/env bash
2+
# Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
3+
#
4+
# Licensed under the Apache License, Version 2.0 (the "License"). You
5+
# may not use this file except in compliance with the License. A copy of
6+
# the License is located at
7+
#
8+
# http://aws.amazon.com/apache2.0/
9+
#
10+
# or in the "license" file accompanying this file. This file is
11+
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
12+
# ANY KIND, either express or implied. See the License for the specific
13+
# language governing permissions and limitations under the License.
14+
15+
./execute_tensorflow_training.py train \
16+
--framework-version 1.12 \
17+
--device gpu \
18+
\
19+
--instance-types ml.p3.16xlarge \
20+
\
21+
--instance-counts 1 \
22+
--instance-counts 2 \
23+
--instance-counts 4 \
24+
--instance-counts 8 \
25+
--instance-counts 16 \
26+
\
27+
--py-versions py3 \
28+
\
29+
--subnets # add subnet id here \
30+
\
31+
--security-groups # add security-group id here

0 commit comments

Comments
 (0)