前言
记录一下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