(self, input)
| 235 | self.fc = layer.Linear(2048, num_classes) |
| 236 | |
| 237 | def features(self, input): |
| 238 | x = self.conv1(input) |
| 239 | x = self.bn1(x) |
| 240 | x = self.relu1(x) |
| 241 | |
| 242 | x = self.conv2(x) |
| 243 | x = self.bn2(x) |
| 244 | x = self.relu2(x) |
| 245 | |
| 246 | x = self.block1(x) |
| 247 | x = self.block2(x) |
| 248 | x = self.block3(x) |
| 249 | x = self.block4(x) |
| 250 | x = self.block5(x) |
| 251 | x = self.block6(x) |
| 252 | x = self.block7(x) |
| 253 | x = self.block8(x) |
| 254 | x = self.block9(x) |
| 255 | x = self.block10(x) |
| 256 | x = self.block11(x) |
| 257 | x = self.block12(x) |
| 258 | |
| 259 | x = self.conv3(x) |
| 260 | x = self.bn3(x) |
| 261 | x = self.relu3(x) |
| 262 | |
| 263 | x = self.conv4(x) |
| 264 | x = self.bn4(x) |
| 265 | return x |
| 266 | |
| 267 | def logits(self, features): |
| 268 | x = self.relu4(features) |