Home Articles

Pytorch中的模型修改

Asked
Viewed 1598 times
0

我想在我保存的CNN模型中更改某些内核中的权重 . 如何更改特定内核中的值并保存到新模型?

1 Answer

  • 0

    你可以 torch.load 你保存的重量 . 你应该得到一个存储权重的 state_dict 字典 . 使用 state_dict 键找到您想要更改的权重,修改它们然后 torch.save 修改后的 state_dict (更好地使用不同的文件名;) .

Related