pranaymanocha / PerceptualAudio

Perceptual Metrics of Audio - perceptually relevant loss function. DPAM and CDPAM

Home Page:https://pixl.cs.princeton.edu/pubs/Manocha_2020_ADP/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cdpam load MELGAN_trained model Error

LXP-Never opened this issue · comments

`import cdpam

loss_fn = cdpam.CDPAM(modfolder="F:/model/cdpam/MELGAN_trained/single_speaker/netG.pt")
wav_ref = cdpam.load_audio('sample_audio/ref.wav') # (1, 55125)
wav_out = cdpam.load_audio('sample_audio/2.wav') # (1, 55125)

dist = loss_fn.forward(wav_ref, wav_out)
print(dist)`

Traceback (most recent call last):
File "F:/model/cdpam/temp.py", line 8, in
loss_fn = cdpam.CDPAM(modfolder="F:/BWE/创新俱乐部/SEANet/model/cdpam/MELGAN_trained/single_speaker/netG.pt")
File "D:\ProgramData\torch_even\lib\site-packages\cdpam\cdpam.py", line 47, in init
state = torch.load(modfolder,map_location="cpu")['state']
KeyError: 'state'