KeyError: ‘state

使用 torch.nn.DataParallel 训练和保存的模型，其 key 中多了’module’，在加载到单GPU或CPU环境中，会报错找不到key，需要将它去掉。

def load_model_without_module(model_state_file):from collections import OrderedDictnew_checkpoint = OrderedDict()checkpoint = torch.load(model_state_file)['state_dict']import pdb; pdb.set_trace()for k, v in checkpoint.items():name = k[7:] # remove module.new_checkpoint[name] = vreturn new_checkpoint
def main():...model = model_state_file = # model = torch.nn.DataParallel(model).cuda()model.load_state_dict(load_model_without_module(model_state_file))

参考：
https://discuss.pytorch.org/t/keyerror-state-dict/18220/2

KeyError: ‘state_dict‘相关推荐

1.无聊，拿GPT2写写诗!
无聊,拿GPT2写写诗! 这个是开源GPT2中文的使用, 安装requirements.txt,一定得安装对应版本torch, transformer, pytorch-lightning,不然会导入 ...
pytorch加载预训练模型遇到的问题:KeyError: ‘bn1.num_batches_tracked‘
问题最近在使用pytorch1.0加载resnet预训练模型时,遇到的一个问题,在此记录一下. KeyError: 'layer1.0.bn1.num_batches_tracked' 其实 ...
pytorch 状态字典:state_dict 模型和参数保存
pytorch 中的 state_dict 是一个简单的python的字典对象,将每一层与它的对应参数建立映射关系.(如model的每一层的weights及偏置等等) (注意,只有那些参数可以训练的l ...
PyTorch加载模型model.load_state_dict()问题，Unexpected key(s) in state_dict: “module.features..，Expected .
希望将训练好的模型加载到新的网络上.如上面题目所描述的,PyTorch在加载之前保存的模型参数的时候,遇到了问题. Unexpected key(s) in state_dict: "mod ...
Python图片处理PIL/pillow/生成验证码/出现KeyError: 和The _imagingft C module is not installed...
近期在用Python开发自己的博客.须要用到Python生成验证码,当然肯定要用到Python的图形处理库PIL,由于我用的是windows. 所以在安装好pil之后就開始写,就依照题目所说出现了Th ...
KeyError: “None of [Int64Index([...],\n dtype=‘int64‘, length=739)] are in the [columns]“
KeyError: "None of [Int64Index([ 0, 1, 2, 3, 4, 6, 7, 8, 9, 10,\n ...
pandas KeyError [‘1‘] not found in axis 错误的解决方法
pandas KeyError: "[''] not found in axis" 错误的解决方法原因就是没有找到这个行或者列 DataFrame数据 ,行有索引,列有名字.不 ...
解决KeyError: ‘acc‘ 和KeyError: ‘val_acc‘错误
问题描述: 程序运行出现KeyError: 'acc' 和KeyError: 'val_acc'的错误. 解决方法: 在Keras 2.3.0中,矩阵的报告方式已更改为与指定的确切名称相匹配. 如果您 ...
python3 日志检索异常抛出异常 raise KeyError(key)，KeyError: ‘formatters‘
python3 日志检索异常抛出异常 raise KeyError(key),KeyError: 'formatters' 参考文章: (1)python3 日志检索异常抛出异常 raise KeyE ...

KeyError: ‘state_dict‘

KeyError: ‘state_dict‘相关推荐

最新文章

热门文章