def forward(self, x): x = self.c1(x) x = self.stage1(x) x = self.stage2(x) x = self.stage3(x) x = self.stage4(x) x = self.bn1(x) x = M.flatten(x) x = F.dropout(x, 0.4, self.training, False) x = self.fc1(x) return x
def forward(self, x): x = self.c1(x) x = self.stage1(x) x = self.stage2(x) x = self.stage3(x) x = self.stage4(x) x = self.bn1(x) x = M.flatten(x) x = F.dropout(x, 0.4, self.training, False) res = [] for fc in self.fcs: res.append(fc(x)) x = torch.cat(res, dim=1) return x
def forward(self, x): x = self.body(x) x = M.flatten(x) feat = self.fc1(x) return feat