def forward(self, x):
x = self.c1(x)
x = self.maxpool(x)
x = self.stage1(x)
x = self.stage2(x)
x = self.stage3(x)
x = self.stage4(x)
x = M.GlobalAvgPool(x)
x = x.squeeze()
x = self.fc1(x)
return x
def forward(self, x):
inp = x
x = self.bn0(x)
x = self.c0(x)
# print(x.shape)
x = self.c1(x)
se = M.GlobalAvgPool(x)
se = self.se1(se)
se = self.se2(se)
# print(x.shape, se.shape)
x = x * se
x = self.c2(x)
if self.outchn == self.inchn and self.stride == 1:
sc = inp
else:
sc = self.sc(inp)
x = sc + x
return x
