Модификация модели в Pytorch

Question

Я хочу изменить веса в определенных ядрах в моей сохраненной модели CNN. Как я могу изменить значения в определенных ядрах и сохранить в новой модели?

Answer 1

Вы можете torch.load веса, которые вы сохранили. Вы должны получить state_dict словарь, в котором хранятся веса. Используйте клавиши state_dict, чтобы найти веса, которые вы хотите изменить, изменить их, а затем torch.save измененный state_dict (лучше использовать другое имя файла;).