DataParallel layers (multi GPU)

import torch.nn as nn

import torch.nn.parallel

import torch.backends.cudnn as cudnn

model = nn.DataParallel(model)