ALWA reduction seems to use too much GPU resources

SonyCSLParis / pesto

Self-supervised learning for fast pitch estimation

GNU Lesser General Public License v3.0

168 stars 15 forks source link

ALWA reduction seems to use too much GPU resources #6

Closed aRI0U closed 9 months ago

aRI0U commented 10 months ago

maybe gather is involved?

aRI0U commented 9 months ago

Cannot manage to reproduce the issue. On a GTX 1080 CUDA OOM happens when trying to pitch estimate more than 9min of audio in parallel with both argmax and alwa reduction methods