in src/run.py [0:0]
def reduce_mean(gen_loss, clf_loss, tot_loss, accuracy, n_gpu):
with tf.device("/gpu:0"):
for i in range(1, n_gpu):
gen_loss[0] += gen_loss[i]
clf_loss[0] += clf_loss[i]
tot_loss[0] += tot_loss[i]
accuracy[0] += accuracy[i]
gen_loss[0] /= n_gpu
clf_loss[0] /= n_gpu
tot_loss[0] /= n_gpu
accuracy[0] /= n_gpu