我想在我保存的CNN模型中更改某些内核中的权重 . 如何更改特定内核中的值并保存到新模型?
你可以 torch.load 你保存的重量 . 你应该得到一个存储权重的 state_dict 字典 . 使用 state_dict 键找到您想要更改的权重,修改它们然后 torch.save 修改后的 state_dict (更好地使用不同的文件名;) .
torch.load
state_dict
torch.save
1 回答
你可以
torch.load
你保存的重量 . 你应该得到一个存储权重的state_dict
字典 . 使用state_dict
键找到您想要更改的权重,修改它们然后torch.save
修改后的state_dict
(更好地使用不同的文件名;) .