Open SarrocaGSergi opened 1 year ago
Hi, It's hard to know without looking into the changed you applied. I just know it doesn't work with newer pytorch versions but forgot why. Could you point me to a public repo to inspect the commits and the code changed?
def wav2sp(self, x):
# CUDNN does not support half complex numbers for non-power2 windows
# Casting to float32 is a workaround
dtype = x.dtype
x = x.float()
s = spectrogram(x, pad=0, window=self._window.float(), win_length=self._n_fft,
n_fft=self._n_fft, hop_length=self._hop_length,
power=None, normalized=False, return_complex=False)
return torch.view_as_real(s)
will work. main reason is the different version of torch.
After adjusting the code to make it run, I find a dimensional error on the forward pass:
This error comes either in the colab notebook and when I clone the repo in my local. The main changes I did is to reformat the requirements to use newer versions of pytorch packages and cuda and fix the bugs generated by
np.int
What do you suggest?