import torch.nn as nn import torch.nn.parallel import torch.backends.cudnn as cudnn model = nn.DataParallel(model)