【pytorch】【net】加载预定义网络googlnet，未导入权重模式时多出现两个附加层

最新推荐文章于 2022-01-06 11:28:16 发布

「已注销」

最新推荐文章于 2022-01-06 11:28:16 发布

阅读量146

点赞数

分类专栏： torch

本文链接：https://blog.csdn.net/YJYS_ZHX/article/details/111593005

版权

一、加载预定义网络
加载googlenet时，是否加载权重对应的网络不太一样。

    net1 = googlenet(pretrained=True)
    net2 = googlenet(pretrained=False)

    print(net1)
    print("*****************")
    print(net2)
    
    #output：net1
    (branch4): Sequential(
      (0): MaxPool2d(kernel_size=3, stride=1, padding=1, dilation=1, ceil_mode=True)
      (1): BasicConv2d(
        (conv): Conv2d(832, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): BatchNorm2d(128, eps=0.001, momentum=0.1, affine=True, track_running_stats=True)
      )
    )
  )
  (aux1): None
  (aux2): None