Multi-GPU Training in Pytorch