mnist-nn-data-parallelism.job

#!/bin/bash

# Change p1p1 to the network interface you want.
mpirun -v --hostfile hostfile --mca btl_tcp_if_include p1p1 \
       -x MNISTNN_PARALLEL="yes" \
       python mnist-nn-data-parallelism.py

#mpirun -v --hostfile hostfile  --mca btl_tcp_if_include p1p1 \
#       -x MNISTNN_PARALLEL="yes" -x MNISTNN_GPU="yes" \
#       python mnist-nn-data-parallelism.py