(self, x)
| 184 | self.fc_layer = nn.Linear(self.feat_dim, self.num_classes) |
| 185 | |
| 186 | def forward(self, x): |
| 187 | feat = self.feature(x) |
| 188 | feat = self.output_layer(feat) |
| 189 | feat = feat.view(feat.size(0), -1) |
| 190 | out = self.fc_layer(feat) |
| 191 | return feat, out |
| 192 | |
| 193 | |
| 194 | class IR152_vib(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected