How to use my own trained model from voicefixer_main?

[utahboy3502]

Hello.

I am having issue when running your code for inference with the trained model from voicefixer_main, not the pretrained model. Is it possible to use the trained model for test.py?

I tried to replaced the vf.ckpt with my trined model ~ at the original directory, but it did not work it produced the following error:

It seems like the pretrained model voicefixer and the trained model from voicefixer_main are different each other in terms of model's size. the pretrained model is about 489.3 MB the one from voice_fixer main is about 1.3 GB

[haoheliu]

Hi, it should work with the model trained from voicefixer_main. But a little tweak in the training code might also be needed. The figure seems to indicate an importing error. The "tools" is a part of this repo and please make sure it can be found and imported by the python interpreter. Many thanks.