DataParallel layers (multi GPU)

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

>>阅读原文<<