search

santi-pdp / segan_pytorch

Speech Enhancement Generative Adversarial Network in PyTorch
MIT License
373 stars 109 forks source link

Reproducible script for training GSEGAN? #35

Open ws-choi opened 2 years ago

ws-choi commented 2 years ago

Hi @santi-pdp, thank you for open-sourcing this amazing project.

I am trying to reproduce training GSEGAN [1] from scratch. However, I could not have found a reproducible script for it. Could you let me know such a train script?

Since there are a lot of parameters and scripts, I am not sure which options I must select to reproduce it. For example, I don't know which dataset I must select.

https://github.com/santi-pdp/segan_pytorch/blob/6b831de85eb7b63e354b7355185f8c70ffb6bfc9/train_gsegan.py#L137-L159

Could you let me know such a train script, similar to this one (for GSEGAN)?

python train.py --save_path ckpt_segan+ --batch_size 300 \
        --clean_trainset data/clean_trainset \
        --noisy_trainset data/noisy_trainset \
        --cache_dir data/cache

Best, Woosung Choi [1] Pascual, Santiago, Joan Serrà, and Antonio Bonafonte. "Towards generalized speech enhancement with generative adversarial networks." arXiv preprint arXiv:1904.03418 (2019). Interspeech

ws-choi commented 2 years ago

Also, can you let me know how I can reproduce the evaluation for GSEGAN?