in evaluations/inception_v3.py [0:0]
def __init__(self, in_channels):
super().__init__()
self.tower = torch.nn.Sequential(
collections.OrderedDict(
[
("conv", Conv2dLayer(in_channels, 192, kh=1, kw=1)),
("conv_1", Conv2dLayer(192, 320, kh=3, kw=3, stride=2)),
]
)
)
self.tower_1 = torch.nn.Sequential(
collections.OrderedDict(
[
("conv", Conv2dLayer(in_channels, 192, kh=1, kw=1)),
("conv_1", Conv2dLayer(192, 192, kh=1, kw=7, padding=[0, 3])),
("conv_2", Conv2dLayer(192, 192, kh=7, kw=1, padding=[3, 0])),
("conv_3", Conv2dLayer(192, 192, kh=3, kw=3, stride=2)),
]
)
)
self.pool = torch.nn.MaxPool2d(kernel_size=3, stride=2)