()
| 69 | |
| 70 | |
| 71 | def test_bitlinear_reset_parameters(): |
| 72 | layer = BitLinear(10, 20) |
| 73 | original_weights = layer.weight.clone() |
| 74 | layer.reset_parameters() |
| 75 | assert not torch.equal(original_weights, layer.weight) |
| 76 | |
| 77 | |
| 78 | @pytest.mark.parametrize("groups", [1, 2, 5]) |