目录
正常加载模型,cpu加载
加载预训练模型pop参数
state_dict = torch.load(weight_path)
state_dict.pop("denoiser.PPE.pe")
self.load_state_dict(state_dict,strict=False)
map_location=torch.device("cpu")
model = IResNet(IBasicBlock, [2, 2, 2, 2])
a_path=r"ms1mv3_arcface_r18_fp16/backbone.pth"
state_dict= torch.load(a_path, map_location=torch.device("cpu"))
model.load_state_dict(state_dict)
自定义加载模型,舍弃某些层:
state_dict = torch.load("220