Блог

Я хотел бы использовать обрезанную модель и вес, но всякий раз, когда я загружаю модель или вес, появляются сообщения об ошибках… Мой вес и данные модели:

Сообщения об ошибках

Причина, по которой у вас возникла эта ошибка, заключается в том, что некоторые веса, присутствующие в файле, который вы пытаетесь загрузить, не имеют соответствующих частей в инициализированной модели (поскольку она была обрезана). Один из быстрых способов обойти это-отключить предупреждения с strict=False помощью опции:

 model.load_state_dict(state_dict, strict=False)
 

На самом деле вы можете сохранить результат этого вызова и зарегистрировать отсутствующие и неожиданные модули, найденные в файле.