AttributeError: ‘XXX‘ object has no attribute ‘module‘

训练模型的时候如果用的是

torch.nn.DataParallel

封装的模型

那么请设置多GPU

CUDA = '0,1'
os.environ['CUDA_VISIBLE_DEVICES'] = CUDA

你可能感兴趣的:(深度学习,各类报错)