├── .gitignore ├── LICENSE ├── README.md ├── assets └── pipeline.png ├── config ├── acc │ ├── fp16_gpus1.yaml │ └── fp16_gpus8.yaml ├── dataset │ ├── dnr.yaml │ └── mix_6k.yaml ├── debug.yaml ├── default.yaml ├── discriminator │ └── default.yaml ├── model │ ├── sdcodec_16k.yaml │ ├── sdcodec_16k_pretrainDAC.yaml │ ├── sdcodec_16k_shard4.yaml │ └── sdcodec_16k_shard8.yaml ├── run_config │ ├── slurm_1.yaml │ └── slurm_debug.yaml └── training │ ├── bs2_iter5k.yaml │ └── bs8_iter400k.yaml ├── environment.yml ├── eval_dnr.py ├── install_visqol.md ├── main.py ├── prepare ├── mani_dnr.py ├── mani_dns_clean.py ├── mani_dns_noise.py ├── mani_jamendo.py ├── mani_musan.py └── mani_wham.py └── src ├── datasets ├── __init__.py └── audio_dataset.py ├── metrics ├── __init__.py ├── adv.py ├── sdr.py ├── spectrum.py └── visqol.py ├── models ├── __init__.py ├── discriminator.py └── sdcodec.py ├── modules ├── __init__.py ├── base_dac.py ├── layers.py └── quantize.py ├── optim ├── __init__.py ├── cosine_lr_scheduler.py ├── exponential_lr_scheduler.py ├── inverse_sqrt_lr_scheduler.py ├── linear_warmup_lr_scheduler.py ├── polynomial_decay_lr_scheduler.py └── reduce_plateau_lr_scheduler.py ├── trainer.py └── utils ├── __init__.py ├── audio_process.py ├── torch_utils.py ├── utils.py └── vis.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/README.md -------------------------------------------------------------------------------- /assets/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/assets/pipeline.png -------------------------------------------------------------------------------- /config/acc/fp16_gpus1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/acc/fp16_gpus1.yaml -------------------------------------------------------------------------------- /config/acc/fp16_gpus8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/acc/fp16_gpus8.yaml -------------------------------------------------------------------------------- /config/dataset/dnr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/dataset/dnr.yaml -------------------------------------------------------------------------------- /config/dataset/mix_6k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/dataset/mix_6k.yaml -------------------------------------------------------------------------------- /config/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/debug.yaml -------------------------------------------------------------------------------- /config/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/default.yaml -------------------------------------------------------------------------------- /config/discriminator/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/discriminator/default.yaml -------------------------------------------------------------------------------- /config/model/sdcodec_16k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/model/sdcodec_16k.yaml -------------------------------------------------------------------------------- /config/model/sdcodec_16k_pretrainDAC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/model/sdcodec_16k_pretrainDAC.yaml -------------------------------------------------------------------------------- /config/model/sdcodec_16k_shard4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/model/sdcodec_16k_shard4.yaml -------------------------------------------------------------------------------- /config/model/sdcodec_16k_shard8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/model/sdcodec_16k_shard8.yaml -------------------------------------------------------------------------------- /config/run_config/slurm_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/run_config/slurm_1.yaml -------------------------------------------------------------------------------- /config/run_config/slurm_debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/run_config/slurm_debug.yaml -------------------------------------------------------------------------------- /config/training/bs2_iter5k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/training/bs2_iter5k.yaml -------------------------------------------------------------------------------- /config/training/bs8_iter400k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/config/training/bs8_iter400k.yaml -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/environment.yml -------------------------------------------------------------------------------- /eval_dnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/eval_dnr.py -------------------------------------------------------------------------------- /install_visqol.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/install_visqol.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/main.py -------------------------------------------------------------------------------- /prepare/mani_dnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/prepare/mani_dnr.py -------------------------------------------------------------------------------- /prepare/mani_dns_clean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/prepare/mani_dns_clean.py -------------------------------------------------------------------------------- /prepare/mani_dns_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/prepare/mani_dns_noise.py -------------------------------------------------------------------------------- /prepare/mani_jamendo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/prepare/mani_jamendo.py -------------------------------------------------------------------------------- /prepare/mani_musan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/prepare/mani_musan.py -------------------------------------------------------------------------------- /prepare/mani_wham.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/prepare/mani_wham.py -------------------------------------------------------------------------------- /src/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/datasets/__init__.py -------------------------------------------------------------------------------- /src/datasets/audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/datasets/audio_dataset.py -------------------------------------------------------------------------------- /src/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/metrics/__init__.py -------------------------------------------------------------------------------- /src/metrics/adv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/metrics/adv.py -------------------------------------------------------------------------------- /src/metrics/sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/metrics/sdr.py -------------------------------------------------------------------------------- /src/metrics/spectrum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/metrics/spectrum.py -------------------------------------------------------------------------------- /src/metrics/visqol.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/metrics/visqol.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/models/__init__.py -------------------------------------------------------------------------------- /src/models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/models/discriminator.py -------------------------------------------------------------------------------- /src/models/sdcodec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/models/sdcodec.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/modules/__init__.py -------------------------------------------------------------------------------- /src/modules/base_dac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/modules/base_dac.py -------------------------------------------------------------------------------- /src/modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/modules/layers.py -------------------------------------------------------------------------------- /src/modules/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/modules/quantize.py -------------------------------------------------------------------------------- /src/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/optim/__init__.py -------------------------------------------------------------------------------- /src/optim/cosine_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/optim/cosine_lr_scheduler.py -------------------------------------------------------------------------------- /src/optim/exponential_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/optim/exponential_lr_scheduler.py -------------------------------------------------------------------------------- /src/optim/inverse_sqrt_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/optim/inverse_sqrt_lr_scheduler.py -------------------------------------------------------------------------------- /src/optim/linear_warmup_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/optim/linear_warmup_lr_scheduler.py -------------------------------------------------------------------------------- /src/optim/polynomial_decay_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/optim/polynomial_decay_lr_scheduler.py -------------------------------------------------------------------------------- /src/optim/reduce_plateau_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/optim/reduce_plateau_lr_scheduler.py -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/trainer.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/audio_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/utils/audio_process.py -------------------------------------------------------------------------------- /src/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/utils/torch_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /src/utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoyuBIE1994/SDCodec/HEAD/src/utils/vis.py --------------------------------------------------------------------------------