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

Question about cupy and torch #226

Closed FireTracer closed 9 months ago

FireTracer commented 10 months ago

I noticed that in version 3.0, copy was removed and torch replaced it. However, in version 4.1, the torch acceleration has also been removed in torchsig/dataset/synthetic. py. So I'm confused about how to accelerate the generation of datasets now, execute scripts/generate_ Sig53.py takes approximately 80 hours and the GPU is not occupied.

Looking forward to your reply, thanks!