DataParallel并行运算的模型导入失败Error的解决方法——解除DataParallel封装

在之前的一篇博客中提到了Pytorch框架下，模型使用GPU多卡并行运算来解决显存不足的问题，即使用：
Pytorch框架下使用多台GPU的数据并行运算的指令

import torch 
from torch.nn.parallel import DataParallel


if torch.cuda.device_count() > 1:
    model = DataParallel(model)  #数据并行运算