├── .dockerignore ├── .gitattributes ├── .gitignore ├── .pre-commit-config.yaml ├── Dockerfile ├── Dockerfile.dev ├── LICENSE ├── README.md ├── assets ├── comparsion_stats.png └── objective_comparisons.png ├── conf ├── 1gpu.yml ├── ablations │ ├── baseline.yml │ ├── diff-mb.yml │ ├── equal-mb.yml │ ├── no-adv.yml │ ├── no-data-balance.yml │ ├── no-low-hop.yml │ ├── no-mb.yml │ ├── no-mpd-msd.yml │ ├── no-mpd.yml │ └── only-speech.yml ├── base.yml ├── downsampling │ ├── 1024x.yml │ ├── 128x.yml │ ├── 1536x.yml │ └── 768x.yml ├── final │ ├── 16khz.yml │ ├── 24khz.yml │ ├── 44khz-16kbps.yml │ └── 44khz.yml ├── quantizer │ ├── 24kbps.yml │ ├── 256d.yml │ ├── 2d.yml │ ├── 32d.yml │ ├── 4d.yml │ ├── 512d.yml │ ├── dropout-0.0.yml │ ├── dropout-0.25.yml │ └── dropout-0.5.yml └── size │ ├── medium.yml │ └── small.yml ├── dac ├── __init__.py ├── __main__.py ├── compare │ ├── __init__.py │ └── encodec.py ├── model │ ├── __init__.py │ ├── base.py │ ├── dac.py │ └── discriminator.py ├── nn │ ├── __init__.py │ ├── layers.py │ ├── loss.py │ └── quantize.py └── utils │ ├── __init__.py │ ├── decode.py │ └── encode.py ├── docker-compose.yml ├── requirements.txt ├── scripts ├── compute_entropy.py ├── evaluate.py ├── get_samples.py ├── mushra.py ├── organize_daps.py ├── save_test_set.py ├── train.py └── train_no_adv.py ├── setup.py └── tests ├── __init__.py ├── test_cli.py └── test_train.py /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/.dockerignore -------------------------------------------------------------------------------- /.gitattributes: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/Dockerfile -------------------------------------------------------------------------------- /Dockerfile.dev: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/Dockerfile.dev -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/README.md -------------------------------------------------------------------------------- /assets/comparsion_stats.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/assets/comparsion_stats.png -------------------------------------------------------------------------------- /assets/objective_comparisons.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/assets/objective_comparisons.png -------------------------------------------------------------------------------- /conf/1gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/1gpu.yml -------------------------------------------------------------------------------- /conf/ablations/baseline.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/baseline.yml -------------------------------------------------------------------------------- /conf/ablations/diff-mb.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/diff-mb.yml -------------------------------------------------------------------------------- /conf/ablations/equal-mb.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/equal-mb.yml -------------------------------------------------------------------------------- /conf/ablations/no-adv.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/no-adv.yml -------------------------------------------------------------------------------- /conf/ablations/no-data-balance.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/no-data-balance.yml -------------------------------------------------------------------------------- /conf/ablations/no-low-hop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/no-low-hop.yml -------------------------------------------------------------------------------- /conf/ablations/no-mb.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/no-mb.yml -------------------------------------------------------------------------------- /conf/ablations/no-mpd-msd.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/no-mpd-msd.yml -------------------------------------------------------------------------------- /conf/ablations/no-mpd.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/no-mpd.yml -------------------------------------------------------------------------------- /conf/ablations/only-speech.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/ablations/only-speech.yml -------------------------------------------------------------------------------- /conf/base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/base.yml -------------------------------------------------------------------------------- /conf/downsampling/1024x.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/downsampling/1024x.yml -------------------------------------------------------------------------------- /conf/downsampling/128x.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/downsampling/128x.yml -------------------------------------------------------------------------------- /conf/downsampling/1536x.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/downsampling/1536x.yml -------------------------------------------------------------------------------- /conf/downsampling/768x.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/downsampling/768x.yml -------------------------------------------------------------------------------- /conf/final/16khz.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/final/16khz.yml -------------------------------------------------------------------------------- /conf/final/24khz.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/final/24khz.yml -------------------------------------------------------------------------------- /conf/final/44khz-16kbps.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/final/44khz-16kbps.yml -------------------------------------------------------------------------------- /conf/final/44khz.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/final/44khz.yml -------------------------------------------------------------------------------- /conf/quantizer/24kbps.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/24kbps.yml -------------------------------------------------------------------------------- /conf/quantizer/256d.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/256d.yml -------------------------------------------------------------------------------- /conf/quantizer/2d.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/2d.yml -------------------------------------------------------------------------------- /conf/quantizer/32d.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/32d.yml -------------------------------------------------------------------------------- /conf/quantizer/4d.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/4d.yml -------------------------------------------------------------------------------- /conf/quantizer/512d.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/512d.yml -------------------------------------------------------------------------------- /conf/quantizer/dropout-0.0.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/dropout-0.0.yml -------------------------------------------------------------------------------- /conf/quantizer/dropout-0.25.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/dropout-0.25.yml -------------------------------------------------------------------------------- /conf/quantizer/dropout-0.5.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/quantizer/dropout-0.5.yml -------------------------------------------------------------------------------- /conf/size/medium.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/size/medium.yml -------------------------------------------------------------------------------- /conf/size/small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/conf/size/small.yml -------------------------------------------------------------------------------- /dac/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/__init__.py -------------------------------------------------------------------------------- /dac/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/__main__.py -------------------------------------------------------------------------------- /dac/compare/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dac/compare/encodec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/compare/encodec.py -------------------------------------------------------------------------------- /dac/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/model/__init__.py -------------------------------------------------------------------------------- /dac/model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/model/base.py -------------------------------------------------------------------------------- /dac/model/dac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/model/dac.py -------------------------------------------------------------------------------- /dac/model/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/model/discriminator.py -------------------------------------------------------------------------------- /dac/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/nn/__init__.py -------------------------------------------------------------------------------- /dac/nn/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/nn/layers.py -------------------------------------------------------------------------------- /dac/nn/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/nn/loss.py -------------------------------------------------------------------------------- /dac/nn/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/nn/quantize.py -------------------------------------------------------------------------------- /dac/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/utils/__init__.py -------------------------------------------------------------------------------- /dac/utils/decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/utils/decode.py -------------------------------------------------------------------------------- /dac/utils/encode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/dac/utils/encode.py -------------------------------------------------------------------------------- /docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/docker-compose.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/compute_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/compute_entropy.py -------------------------------------------------------------------------------- /scripts/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/evaluate.py -------------------------------------------------------------------------------- /scripts/get_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/get_samples.py -------------------------------------------------------------------------------- /scripts/mushra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/mushra.py -------------------------------------------------------------------------------- /scripts/organize_daps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/organize_daps.py -------------------------------------------------------------------------------- /scripts/save_test_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/save_test_set.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/train_no_adv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/scripts/train_no_adv.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/tests/test_cli.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/descriptinc/descript-audio-codec/HEAD/tests/test_train.py --------------------------------------------------------------------------------