Error(s) in loading state_dict for DataParallel

关于PyTorch模型保存与导入的一些注意点：python 1.没有使用并行计算：函数 import torch.nn as nn class Net(nn.Module): def __init__(self): super().__init__() self.conv1 = nn.Conv2d(2, 2, 1) self.linear

>>阅读原文<<