()
| 4 | |
| 5 | |
| 6 | def test_bitlinear_initialization(): |
| 7 | bitlinear = BitLinear(in_features=512, out_features=256, bias=True) |
| 8 | assert bitlinear.in_features == 512 |
| 9 | assert bitlinear.out_features == 256 |
| 10 | assert bitlinear.weight.shape == (256, 512) |
| 11 | assert bitlinear.bias.shape == (256,) |
| 12 | assert bitlinear.gamma.shape == (512,) |
| 13 | assert bitlinear.beta.shape == (256,) |
| 14 | |
| 15 | |
| 16 | def test_bitlinear_forward_pass(): |