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'