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