(self, d2, d1, t1, t2, d3, t3, d4, t4)
| 44 | |
| 45 | class BiN_CTABL(pl.LightningModule): |
| 46 | def __init__(self, d2, d1, t1, t2, d3, t3, d4, t4): |
| 47 | super().__init__() |
| 48 | |
| 49 | self.BiN = BiN(d2, d1, t1, t2) |
| 50 | self.BL = BL_layer(d2, d1, t1, t2) |
| 51 | self.BL2 = BL_layer(d3, d2, t2, t3) |
| 52 | self.TABL = TABL_layer(d4, d3, t3, t4) |
| 53 | self.dropout = nn.Dropout(0.1) |
| 54 | |
| 55 | def forward(self, x): |
| 56 | x = x.squeeze(1) |
nothing calls this directly
no test coverage detected