| 194 | |
| 195 | |
| 196 | class ClassifierModule(nn.Module): |
| 197 | def __init__(self, m, channel, num_classes): |
| 198 | super(ClassifierModule, self).__init__() |
| 199 | self.m = m |
| 200 | self.linear = nn.Linear(channel, num_classes) |
| 201 | |
| 202 | def forward(self, x): |
| 203 | res = self.m(x[-1]) |
| 204 | res = res.view(res.size(0), -1) |
| 205 | return self.linear(res), res |
| 206 | |
| 207 | class MSDNet(nn.Module): |
| 208 | def __init__(self, args): |
no outgoing calls
no test coverage detected