(self, x)
| 227 | self.linear = nn.Linear(channel, num_classes) |
| 228 | |
| 229 | def forward(self, x): |
| 230 | res = self.m(x[-1]) |
| 231 | res = res.view(res.size(0), -1) |
| 232 | return self.linear(res), res |
| 233 | |
| 234 | class MSDNet(nn.Module): |
| 235 | def __init__(self, args): |
nothing calls this directly
no outgoing calls
no test coverage detected