'Sequential' 对象没有属性 'classifier'

Question

如何找到in_features的Pytorch模型？ model.classifier.in_features 在 densenet121 上工作，但在 vgg18 上运行 nit，是否有任何功能可以在所有 torchvision 模型上工作？

Answer 1

classifier 是 VGG 实现中的一个 Sequential 模块，因此，如果您想访问传递给 classifier 的 in_features ，你要勾选第一层的in_features

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

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