(self, x)
| 160 | self.fc_layer = nn.Linear(self.feat_dim, self.num_classes) |
| 161 | |
| 162 | def forward(self, x): |
| 163 | feat = self.feature(x) |
| 164 | feat = self.output_layer(feat) |
| 165 | feat = feat.view(feat.size(0), -1) |
| 166 | out = self.fc_layer(feat) |
| 167 | __, iden = torch.max(out, dim=1) |
| 168 | iden = iden.view(-1, 1) |
| 169 | return feat, out |
| 170 | |
| 171 | |
| 172 | class IR152(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected