Как решить проблему несоответствия размеров при загрузке предварительно обученной модели в pytorch? - PullRequest
0 голосов
/ 04 апреля 2020

Я пытаюсь загрузить предварительно обученную модель, в которой количество фильтров (каналов) конвективных слоев составляет 64, в модель с 128 фильтрами в конвективных слоях. Я использую new_network.load_state_dict(pretrained_model, strict=False), но я сталкиваюсь с ошибкой несоответствия размера для всех этих слоев. Как я могу инициализировать некоторые из фильтров в каждом слое сверления с доступными предварительно подготовленными весами модели?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...