GoogLeNet框架
Welcome file import torch import torch . nn as nn import torch . nn . functional as F from torch . nn . modules . module import T from torch . utils . hooks import RemovableHandle class InceptionModule ( nn . Module ) : def __init__ ( self , in_channels , ch1x1 , ch3x3red , ch3x3 , ch5x5red , ch5x5 , pool_proj ) : super ( ) . __init__ ( ) # 分支1:1x1卷积 self . branch1 = nn . Conv2d ( in_channels , ch1x1 , kernel_size = 1 ) # 分支2:1x1 -> 3x3卷积 self . branch2 = nn . Sequential ( nn . Conv2d ( in_channels , ch3x3red , kernel_size = 1 ) , nn . Conv2d ( ch3x3red , ch3x3 , kernel_size = 3 , padding = 1 ) ) # 分支3:1x1 -> 5x5 卷积 self . branch3 = nn . Sequential ( nn . Conv2d ( in_channels , ch5x5red , kernel_size = 1 ) , nn . Conv2d ( ch5x5red , ch5x5 , kernel_size = 5 , padding = 2 ) ) # 分支4:3x3 池化 -> 1x1 卷积 self . branch4 = nn . Sequential ( nn . MaxPool2d ( kernel_size = 3 , stride = 1...