search

RookieJunChen / FullSubNet-plus

The official PyTorch implementation of "FullSubNet+: Channel Attention FullSubNet with Complex Spectrograms for Speech Enhancement".
Apache License 2.0
242 stars 55 forks source link

failed to load pre-trained model #7

Closed fromprimary closed 2 years ago

fromprimary commented 2 years ago

I tried to make a quick usage using pre-trained checkpoint, but get the error "magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: A load persistent id instruction was encountered, but no persistent_load function was specified."
my torch is 1.7.1 ,python is 3.8 . The command is "python -m speech_enhance.tools.inference -C config/inference.toml -M archive/data.pkl -I ~/mini_test_set/test.wav -O ~/fullout" What can I do to run the demo successfully?

fromprimary commented 2 years ago

do not unzip the pre-trained model