Pytorch加载模型时报错 RuntimeError: Error(s) in loading state_dict for DataParallel 问题的一种解决办法
Pytorch加载模型时报错:此时的加载部分代码为: model = torch.nn.DataParallel(resnet.__dict__[args.arch]()) model.load_state_dict(torch.load(args.weights)) model.cuda()修改第二行代码为: model = torch.nn.DataParallel(resnet.__dict__[args.arch]()) model.load_state_d
原创
2021-03-12 21:57:03 ·
2083 阅读 ·
1 评论