加载checkpoints时报错没有键‘model’

问题：

        最近遇到一个错误，在使用torch.load加载.pth预训练模型时，想提取其中键为'model'的value，报错keyError

解决方法：

        打印checkpoints确实没有这个键，而是包含了模型的各个层参数，这个预训练模型是OrderedDict，因此可以直接把checkpoint做为模型参数，代码改为

 注意：

       由于是OrderedDict ，之后如果要访问state_dict里面的键，不能直接用state_dict['xxx']的方式访问，需要通过索引或者迭代的方式访问键值对

for key, value in state_dict.items():
    if 'visual_encoder.pos_embed' in key:
        pos_embed_reshaped = interpolate_pos_embed(value, model.visual_encoder)
        state_dict[key] = pos_embed_reshaped