(self, d2, d1, t1, t2, d3, t3)
| 8 | |
| 9 | class BiN_BTABL(pl.LightningModule): |
| 10 | def __init__(self, d2, d1, t1, t2, d3, t3): |
| 11 | super().__init__() |
| 12 | |
| 13 | self.BiN = BiN(d2, d1, t1, t2) |
| 14 | self.BL = BL_layer(d2, d1, t1, t2) |
| 15 | self.TABL = TABL_layer(d3, d2, t2, t3) |
| 16 | self.dropout = nn.Dropout(0.1) |
| 17 | |
| 18 | def forward(self, x): |
| 19 | x = x.squeeze(1) |
no test coverage detected