TorchDSP / torchsig

TorchSig is an open-source signal processing machine learning toolkit based on the PyTorch data handling pipeline.
MIT License
155 stars 37 forks source link

Quantize() to have broader range of levels #213

Closed MattCarrickPL closed 3 weeks ago

MattCarrickPL commented 11 months ago

Quantize() transform currently uses 16, 24, 32, 40, 48, 56, 64 levels which are all represented by 4-6 bits.

def __init__(
    self,
    num_levels: IntParameter = UniformDiscreteRD(
        np.asarray([16, 24, 32, 40, 48, 56, 64], dtype=int)
    ),
    round_type: List[str] = (["floor", "middle", "ceiling"]),
) -> None:

https://github.com/TorchDSP/torchsig/blob/v0.4.2/torchsig/transforms/transforms.py#L2961

Would like to increase the quantization range to include from 7 to 16 bits.

MattCarrickPL commented 3 weeks ago

Old issue, closing out