(self, *, backbone: nn.Module, linear_head: nn.Module, layers: int = 4)
| 51 | |
| 52 | class _LinearClassifierWrapper(nn.Module): |
| 53 | def __init__(self, *, backbone: nn.Module, linear_head: nn.Module, layers: int = 4): |
| 54 | super().__init__() |
| 55 | self.backbone = backbone |
| 56 | self.linear_head = linear_head |
| 57 | self.layers = layers |
| 58 | |
| 59 | def forward(self, x): |
| 60 | if self.layers == 1: |
nothing calls this directly
no outgoing calls
no test coverage detected