Pytorch distributed elastic: Socket Timeout

Double post from here.