RuntimeError: Error(s) in loading state_dict for & size mismatch for

RuntimeError: Error(s) in loading state_dict for & size mismatch for_第1张图片

 出现这种错误是因为你使用的预训练模型是使用多GPU训练的,所以只要在模型加载前加上一句

model = nn.DataParallel(model)//cpu环境

如果是GPU环境,则

model = nn.DataParallel(model).cuda()

然后再跑模型就能正常训练了。

你可能感兴趣的:(错误,深度学习,python)