Skip to content

Commit fb0a3ed

Browse files
mvsuspEliza Zhang
authored andcommitted
Add Horovod tests (aws#151)
1 parent 5106c5a commit fb0a3ed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/resources/mnist/horovod_mnist.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
import tensorflow as tf
2525
import horovod.keras as hvd
2626

27+
2728
# Horovod: initialize Horovod.
2829
hvd.init()
2930

0 commit comments

Comments
 (0)