【发布时间】：2020-04-22 14:09:39
【问题描述】：
如何找到 Pytorch 模型的 in_features？ model.classifier.in_features 在densenet121 上工作，但在vgg18 上工作，是否有任何功能可以在所有torchvision 模型上工作？
【问题讨论】：
标签： opencv pytorch artificial-intelligence conv-neural-network torchvision
【发布时间】：2020-04-22 14:09:39
【问题描述】：
classifier是VGG的实现中的Sequential模块，所以，如果你想访问传递给classifier的in_features，你必须检查第一个的in_features层。
models.vgg19().classifier[0].in_features
似乎不同的实现遵循不同的模式，因此为所有模型确定in_features 的最佳方法是直接检查源代码。
【讨论】：