(self, input)
| 200 | self.classifier = nn.Linear(C_prev, num_classes) |
| 201 | |
| 202 | def forward(self, input): |
| 203 | logits_aux = None |
| 204 | s0 = self.stem0(input) |
| 205 | s1 = self.stem1(s0) |
| 206 | for i, cell in enumerate(self.cells): |
| 207 | s0, s1 = s1, cell(s0, s1, self.drop_path_prob) |
| 208 | if i == 2 * self._layers // 3: |
| 209 | if self._auxiliary and self.training: |
| 210 | logits_aux = self.auxiliary_head(s1) |
| 211 | out = self.global_pooling(s1) |
| 212 | logits = self.classifier(out.view(out.size(0), -1)) |
| 213 | return logits, logits_aux |
| 214 |
nothing calls this directly
no outgoing calls
no test coverage detected