Run example training on yarn failed.

[luoli523]

Guys:
I am trying to launch example traing_mnist.py on Yarn cluster which with 50 nodes, the launch script like below:

${mxnethome}/tools/launch.py -n 2 \
      --launcher yarn \
     python ${mxnethome}/example/image-classification/train_mnist.py --network lenet --kv-store dist_async

It keeps failed and the yarn container told me below logs:

File "./train_mnist.py", line 1, in <module>
    import find_mxnet
ImportError: No module named find_mxnet

and there is no method to add -file option to the script of tools/launch.py?

Any one run the distributed training example on Yarn successfully?

[avspavan]

I have the same issue, can anyone help here?