def bit_rate(self) -> int:
    return {SparseType.FP32.value: 32, SparseType.FP16.value: 16, SparseType.INT8.value: 8, SparseType.INT4.value: 4, SparseType.INT2.value: 2}[self.value]
def bit_rate(self) -> int:
    return {SparseType.FP32.value: 32, SparseType.FP16.value: 16, SparseType.INT8.value: 8, SparseType.INT4.value: 4, SparseType.INT2.value: 2}[self.value]