Pytorch中的模型修改

Question

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

Answer 1

您可以torch.load保存重量。您应该得到一个state_dict字典，其中存储了权重。使用state_dict键找到要更改的权重，然后进行修改，然后torch.save修改后的state_dict（最好使用不同的文件名;）。