Pytorch - вывод линейного слоя in_features

Question

Я строю игрушечную модель, чтобы взять несколько изображений и дать мне классификацию.Моя модель выглядит так:

conv2d -> pool -> conv2d -> linear -> linear.

Моя проблема заключается в том, что при создании модели мы должны рассчитать размер первого линейного слоя in_features на основе размеравходное изображение.Если мы получаем новые изображения разных размеров, мы должны пересчитать in_features для нашего линейного слоя.Почему мы должны это сделать?Разве это не может быть просто выведено?

Answer 1

Почему вы ожидаете, что линейный слой выведет свой входной размер?Что делать, если вы намеренно захотите изменить этот размер (т. Е. Выходные каналы для слоя конвона или что-то еще).Я считаю, что ваша работа должна быть параметризована (то есть контролироваться четко определенными параметрами).Обходной путь должен всегда преобразовывать входные изображения в определенную форму.Но я не рекомендую так, поскольку это может сильно повлиять на точность вашей модели, потому что преобразование включает в себя потерю пикселей (т. Е. Особенности вашего ввода).