问题描述
1、Missing key(s) in state_dict
2、测试结果全部为0或者全部为某一固定值
解决方法:
checkpoint = torch.load(path)
model.load_state_dict({k.replace(‘module.’, ‘’): v for k, v in checkpoint.items()}, strict=False)
问题描述
1、Missing key(s) in state_dict
2、测试结果全部为0或者全部为某一固定值
解决方法:
checkpoint = torch.load(path)
model.load_state_dict({k.replace(‘module.’, ‘’): v for k, v in checkpoint.items()}, strict=False)