Reuse Pytorch GoogLeNet Structure
作者:XD / 发表: 2020年11月13日 03:46 / 更新: 2020年11月13日 03:46 / 编程笔记 / 阅读量:1994
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])