(self)
| 20 | |
| 21 | class final_model(nn.Module): # Total parameters: 158.64741325378418 MB |
| 22 | def __init__(self): |
| 23 | super(final_model, self).__init__() |
| 24 | |
| 25 | self.convnext = convnext_base(num_classes=2) |
| 26 | self.convnext = augment_inputs_network(self.convnext) |
| 27 | |
| 28 | self.replknet = create_RepLKNet31B(num_classes=2) |
| 29 | self.replknet = augment_inputs_network(self.replknet) |
| 30 | |
| 31 | def forward(self, x): |
| 32 | B, N, C, H, W = x.shape |
no test coverage detected