Skip to content

Commit e193ae1

Browse files
committed
fix horovodrun command in unit test
1 parent 8239008 commit e193ae1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/examples/test_multi_GPU_with_horovod_and_tensorflow.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@ def test_getting_started_tensorflow(tb, tmpdir):
4949
hvd_wrap_path,
5050
"python",
5151
"tf_trainer.py",
52+
f"--data_path={str(tmpdir)}",
53+
"--batch_size=65536",
5254
],
5355
stdout=subprocess.PIPE,
5456
stderr=subprocess.PIPE,

0 commit comments

Comments
 (0)