EADST

Reuse Pytorch GoogLeNet Structure

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])
相关标签
About Me
XD
Goals determine what you are going to be.
Category
标签云
站点统计

本站现有博文266篇,共被浏览440583

本站已经建立2019天!

热门文章
文章归档
回到顶部