【功能代码分享】模型参数非严格性迁移
能够将E模型参数迁移到D模型中,即使两者的键不同。
def load_fortune_model(params, model):
from tqdm import tqdm
checkpoint = torch.load(params['pretrained'], map_location='cpu')
try:
D_dict = model.module.state_dict()
except AttributeError:
D_dict
原创
2021-10-10 15:35:20 ·
123 阅读 ·
0 评论