Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Where can I download the AAS pre-trained model? #4

Open
YounggjuuChoi opened this issue Oct 9, 2019 · 2 comments
Open

Where can I download the AAS pre-trained model? #4

YounggjuuChoi opened this issue Oct 9, 2019 · 2 comments

Comments

@YounggjuuChoi
Copy link

I want to test the AAS.
And the test command line is shown below.

python main.py --trainer AAS --DB_name chime --rnn_size 500 --rnn_layers 4 --ASR_path ../AM_training/models/librispeech_final.pth.tar --mode test --load_path /data/kenkim/AAS_enhancement/model.pth.tar

So, I want to obtain the file 'model.pth.tar'.
Where can I download the file?

@lifelongeek
Copy link
Owner

Thanks for your interests on AAS.
Check the readme page.

"The pre-trained enhanement model (E) trained on Librispeech + DEMAND is available here. (acoustic:adversarial = 1:100000, #hidden = 500, #layer = 4)"

@YounggjuuChoi
Copy link
Author

Thank you for your answer.

I have one more question.

In 'main.py' code, there are the following parts.

if (config.mode == 'train'):
    trainer.train() # VAE
elif(config.mode == 'test'):
    trainer.test()
elif(config.mode == 'visualize'):
    trainer.visualize()

But, in 'trainer_AAS.py' code, the test() function could not be found.

How can I test the AAS?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants