horovodrun -np 1 -H localhost:1 \ python train_re.py --config config/hps-refcoco+_conceptual_rank.json