“‘Sequential”对象没有属性“classifier”

Question

如何找到Pytorch模型的in_features？model.classifier.in_features在densenet121上工作，而nit在vgg18上工作，有没有什么函数可以在所有的torchvision模型上工作？

Answer 1

classifier是VGG实现中的一个Sequential模块，所以，如果你想访问传递给classifier的in_features，你必须检查第一层的in_features。

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

似乎不同的实现遵循不同的模式，因此确定所有模型的in_features的最好方法是直接检查源代码。