You can get it out like so: model = model.module where model was initially wrapped by DataParallel.
It’s kind of annoying because, depending on your code, you might have to have if statements surrounding it. (eg., if using single gpu, don’t do this)