model.load_state_dict({k.replace(‘module.’,’’):v for k,v in torch.load(checkpoint_path)[‘state_dict’].items()})
python保存的模型中有module关键字
最新推荐文章于 2024-05-03 19:06:40 发布
model.load_state_dict({k.replace(‘module.’,’’):v for k,v in torch.load(checkpoint_path)[‘state_dict’].items()})