Mujtaba Asif

Hi, Thanks for providing the implementation it really helps. I am getting the following error when trying to use Multiple GPUs with DataParallel. Please note the implementation works perfectly fine...