如何查找Pytorch模型的in_features?model.classifier.in_features在densenet121上可以工作,但在vgg18上不行,有没有一个函数可以适用于所有torchvision模型?
回答:
classifier在VGG的实现中是一个Sequential模块,因此,如果你想访问传递给classifier的in_features,你需要检查第一层的in_features。
models.vgg19().classifier[0].in_features
看起来不同的实现遵循不同的模式,因此确定所有模型的in_features的最佳方法是直接检查源代码。