interactiveaudiolab / penn

Pitch Estimating Neural Networks (PENN)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Issue in Running in Colab with GPU, even same on Mac (M2) [MPS]

SutirthaChakraborty opened this issue · comments

import penn 
audio, sample_rate = torchaudio.load('/content/try this.wav')
 
hopsize = .01
 
fmin = 30.
fmax = 1000.
 
gpu = 0
  
batch_size = 2048  
checkpoint = None
 
center = 'half-hop'
 
interp_unvoiced_at = .065
 
pitch, periodicity = penn.from_audio(
    audio,
    sample_rate,
    hopsize=hopsize,
    fmin=fmin,
    fmax=fmax,
    checkpoint=checkpoint,
    batch_size=batch_size,
    center=center,
    interp_unvoiced_at=interp_unvoiced_at,
    gpu=gpu)
image

Thanks for opening the issue. Fixed in 0.0.11