如何查找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
的最佳方法是直接检查源代码。