"""call the distributed process"""

from subprocess import Popen# call shell command

hosts = ['gpu1', 'gpu2', 'gpu3']

hostfile = open('hosts', 'w')
hostfile.write('\n'.join(hosts))
hostfile.close()

command = '''python3 ../../tools/launch.py -n 3 --launcher ssh -H hosts "export MXNET_PROFILER_AUTOSTART=1; python3 train_cifar10.py --num-layers 56 --kv-store dist_sync --gpus 0 --num-epochs 5"'''

proc = Popen(command, shell=True)
proc.wait()