torch Missing key(s) in state_dict

在训练的时候,使用了单机多卡,load 模型的时候出现了问题,搜索之后发现,是因为单机多卡的原因。所以需要在torch.load前面加上多GPU的设置:就OK了
torch Missing key(s) in state_dict_第1张图片

你可能感兴趣的:(torch学习)