Use pickletools.optimize() to prevent OOM during dataset generation.

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

Closed anarkiwi closed 2 months ago

anarkiwi commented 11 months ago

Also pin matplot to 3.7.2 as the latest isn't compatible.

TorchDSP commented 2 months ago

v0.5.2 should fix the OOM issue