SonyCSLParis / pesto

Self-supervised learning for fast pitch estimation
GNU Lesser General Public License v3.0
168 stars 15 forks source link

Chunk gpu #11

Closed aRI0U closed 9 months ago

aRI0U commented 9 months ago

A num_chunks option has been added for enabling splitting batch of CQT frames into chunks that will be then processed sequentially. This enable to limit memory usage hence making it possible to deal with longer sequences.

The default (num_chunks=1) exactly corresponds to the previous behaviour.

As a reference, this splitting technique now enables to process one hour of audio in 20s with a single GTX 1080.