读取模型文件,并保存至有序字典中:
from collections import OrderedDict
new_state_dict = OrderedDict()
for k, v in state_dict.items():
new_state_dict[k] = v
读取模型文件,并保存至有序字典中:
from collections import OrderedDict
new_state_dict = OrderedDict()
for k, v in state_dict.items():
new_state_dict[k] = v