ResNet, Inception: input_size = model.fc.in_features
VGG: input_size = model.classifier[0].in_features
DenseNet: input_size = model.classifier.in_features
SqueezeNet: input_size = model.classifier[1].in_channels
AlexNet: alexnet.classifier[1].in_features
pytorch 各个预训练网络全连接层输入参数查询
最新推荐文章于 2022-05-18 17:13:13 发布