一点关于OrderedDict

前言

记录一下OrderedDict是因为pytorch加载参数(torch.load(<model_path>)['state_dict'])是OrderedDict类型，其中以(key, value)的形式存储着整个模型的参数。作为Dictionary的一个子类，最大不同在于保存了items的插入顺序。具体参考：文档。

直接上例子

测试代码：

from collections import OrderedDict

od = OrderedDict()
od['a'] = 123
od['b'] = 234
print("1: ", od)

od['a'] = 345
print("2: ", od)

if 'a' in od:
    print("3: ", True)
else:
    print("3: ", False)

for i in od:
    print("4: ", od[i])

打印结果：

1:  OrderedDict([('a', 123), ('b', 234)])
2:  OrderedDict([('a', 345), ('b', 234)])
3:  True
4:  345
4:  234