Single click distributed training using Tensorflow