«Последовательный» объект не имеет атрибута «классификатор» - PullRequest
0 голосов
/ 22 апреля 2020

Как найти in_features модели Pytorch? model.classifier.in_features работает на densenet121, но не на vgg18, есть ли функции, которые могут работать на всех моделях torchvision?

1 Ответ

0 голосов
/ 22 апреля 2020

classifier - это модуль Sequential в реализации VGG, поэтому, если вы хотите получить доступ к in_features, переданному classifier, вы должны проверить in_features первого layer.

models.vgg19().classifier[0].in_features

Кажется, что разные реализации следуют разным шаблонам, поэтому лучший способ определить in_features для всех моделей - это проверить исходный код напрямую.

...