Convert tf.argmax results to numpy array

Question

I am new to Tensorflow and wrote the following distributed training code. The code works fine.

import multiprocessing
import os
import portpicker
import tensorflo