Reuse Pytorch GoogLeNet Structure
作者：XD / 发表： 2020年11月13日 03:46 / 更新： 2020年11月13日 03:46 / 编程笔记 / 阅读量：3107
Here is the method to reconstruct the GoogLeNet model with Pytorch.
from torch import nn
import torchvision.models as models
base_model = models.googlenet(pretrained=True)
# close aux 1&2
base_model.aux_logits = False
# extract the model without avgpool and FC layer
base_layers = nn.Sequential(*list(base_model.children())[:-3])
