CNTK single gpu doesn't run because of node count comparison. (#79)

* Bugfix: cntk single gpu is never ran because of node count comparison.

For single node, single gpu scenario, $AZ_BATCH_HOST_LIST has a single
IP, and hence the count of nodes will be 1.

* Updating condition.
This commit is contained in:
Paresh Verma 2017-05-05 19:02:39 -07:00 коммит произвёл Fred Park
Родитель 1050c5da0e
Коммит 6eee2c120a
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -17,7 +17,7 @@ IFS=',' read -ra HOSTS <<< "$AZ_BATCH_HOST_LIST"
nodes=${#HOSTS[@]}
# special path for non-mpi job with single gpu
if [ $nodes -eq 0 ] && [ $ngpus -eq 1 ]; then
if [ $nodes -le 1 ] && [ $ngpus -eq 1 ]; then
echo "running cntk in single node + single gpu mode"
# set cntk file
cntkfile=/cntk/Examples/Image/Classification/ConvNet/BrainScript/ConvNet_MNIST.cntk