├── .github └── workflows │ ├── pytest.yaml │ └── static-gh-pages.yaml ├── .gitignore ├── .gitlab-ci.yml ├── CHANGELOG.md ├── LICENSE ├── Makefile ├── README.md ├── compressai_trainer ├── __init__.py ├── config │ ├── __init__.py │ ├── config.py │ ├── dataset.py │ ├── engine.py │ ├── env.py │ ├── load.py │ └── outputs.py ├── plot │ ├── __init__.py │ ├── distribution.py │ ├── featuremap.py │ ├── pdf_signatures.py │ ├── rd.py │ └── utils.py ├── registry │ ├── __init__.py │ ├── catalyst.py │ ├── torch.py │ └── torchvision.py ├── run │ ├── __init__.py │ ├── eval_model.py │ ├── plot_rd.py │ └── train.py ├── runners │ ├── __init__.py │ ├── base.py │ ├── gvae_image_compression.py │ ├── image_compression.py │ └── utils.py ├── typing │ ├── __init__.py │ ├── catalyst.py │ └── torch.py └── utils │ ├── __init__.py │ ├── aim │ ├── __init__.py │ ├── distribution.py │ ├── image.py │ └── query.py │ ├── args.py │ ├── catalyst │ ├── .gitignore │ ├── __init__.py │ └── loggers │ │ ├── __init__.py │ │ ├── aim.py │ │ └── logger.py │ ├── compressai │ ├── __init__.py │ └── results.py │ ├── git.py │ ├── metrics.py │ ├── optimal.py │ ├── pip.py │ ├── system.py │ └── utils.py ├── conf ├── criterion │ └── RateDistortionLoss.yaml ├── dataset │ ├── clic2020-mobile │ │ ├── infer.yaml │ │ ├── train.yaml │ │ └── valid.yaml │ ├── clic2020-professional │ │ ├── infer.yaml │ │ ├── train.yaml │ │ └── valid.yaml │ ├── kodak │ │ └── infer.yaml │ ├── pregenerated │ │ ├── train.yaml │ │ └── valid.yaml │ └── vimeo90k │ │ ├── train.yaml │ │ └── valid.yaml ├── engine │ └── default.yaml ├── env │ └── default.yaml ├── eval_zoo.yaml ├── example.yaml ├── example_gvae.yaml ├── example_gvae_pregenerated.yaml ├── example_pregenerated.yaml ├── misc │ └── default.yaml ├── optimizer │ └── net_aux.yaml ├── paths │ └── default.yaml ├── runner │ ├── GVAEImageCompressionRunner.yaml │ └── ImageCompressionRunner.yaml └── scheduler │ ├── OneCycleLR.yaml │ └── ReduceLROnPlateau.yaml ├── docs ├── .gitignore ├── Makefile ├── README.md ├── make.bat └── source │ ├── _static │ └── logo.svg │ ├── compressai_trainer │ ├── config.rst │ ├── plot.rst │ ├── registry.rst │ ├── run.rst │ ├── runners.rst │ ├── typing.rst │ └── utils.rst │ ├── conf.py │ ├── index.rst │ ├── media │ └── images │ │ └── aim-run-figure-rd-curves.png │ ├── tools │ ├── eval_model.rst │ ├── rd_plotter.rst │ └── train.rst │ └── tutorials │ ├── full.rst │ └── installation.rst ├── poetry.lock ├── pyproject.toml ├── tests ├── SyntheticDataset.yaml ├── __init__.py ├── test_train.py └── utils.py └── train.py /.github/workflows/pytest.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/.github/workflows/pytest.yaml -------------------------------------------------------------------------------- /.github/workflows/static-gh-pages.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/.github/workflows/static-gh-pages.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitlab-ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/.gitlab-ci.yml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/README.md -------------------------------------------------------------------------------- /compressai_trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/config/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/config/config.py -------------------------------------------------------------------------------- /compressai_trainer/config/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/config/dataset.py -------------------------------------------------------------------------------- /compressai_trainer/config/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/config/engine.py -------------------------------------------------------------------------------- /compressai_trainer/config/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/config/env.py -------------------------------------------------------------------------------- /compressai_trainer/config/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/config/load.py -------------------------------------------------------------------------------- /compressai_trainer/config/outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/config/outputs.py -------------------------------------------------------------------------------- /compressai_trainer/plot/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/plot/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/plot/distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/plot/distribution.py -------------------------------------------------------------------------------- /compressai_trainer/plot/featuremap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/plot/featuremap.py -------------------------------------------------------------------------------- /compressai_trainer/plot/pdf_signatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/plot/pdf_signatures.py -------------------------------------------------------------------------------- /compressai_trainer/plot/rd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/plot/rd.py -------------------------------------------------------------------------------- /compressai_trainer/plot/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/plot/utils.py -------------------------------------------------------------------------------- /compressai_trainer/registry/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/registry/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/registry/catalyst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/registry/catalyst.py -------------------------------------------------------------------------------- /compressai_trainer/registry/torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/registry/torch.py -------------------------------------------------------------------------------- /compressai_trainer/registry/torchvision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/registry/torchvision.py -------------------------------------------------------------------------------- /compressai_trainer/run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/run/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/run/eval_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/run/eval_model.py -------------------------------------------------------------------------------- /compressai_trainer/run/plot_rd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/run/plot_rd.py -------------------------------------------------------------------------------- /compressai_trainer/run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/run/train.py -------------------------------------------------------------------------------- /compressai_trainer/runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/runners/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/runners/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/runners/base.py -------------------------------------------------------------------------------- /compressai_trainer/runners/gvae_image_compression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/runners/gvae_image_compression.py -------------------------------------------------------------------------------- /compressai_trainer/runners/image_compression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/runners/image_compression.py -------------------------------------------------------------------------------- /compressai_trainer/runners/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/runners/utils.py -------------------------------------------------------------------------------- /compressai_trainer/typing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/typing/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/typing/catalyst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/typing/catalyst.py -------------------------------------------------------------------------------- /compressai_trainer/typing/torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/typing/torch.py -------------------------------------------------------------------------------- /compressai_trainer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/utils/aim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/aim/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/utils/aim/distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/aim/distribution.py -------------------------------------------------------------------------------- /compressai_trainer/utils/aim/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/aim/image.py -------------------------------------------------------------------------------- /compressai_trainer/utils/aim/query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/aim/query.py -------------------------------------------------------------------------------- /compressai_trainer/utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/args.py -------------------------------------------------------------------------------- /compressai_trainer/utils/catalyst/.gitignore: -------------------------------------------------------------------------------- 1 | !loggers/ 2 | -------------------------------------------------------------------------------- /compressai_trainer/utils/catalyst/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/catalyst/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/utils/catalyst/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/catalyst/loggers/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/utils/catalyst/loggers/aim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/catalyst/loggers/aim.py -------------------------------------------------------------------------------- /compressai_trainer/utils/catalyst/loggers/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/catalyst/loggers/logger.py -------------------------------------------------------------------------------- /compressai_trainer/utils/compressai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/compressai/__init__.py -------------------------------------------------------------------------------- /compressai_trainer/utils/compressai/results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/compressai/results.py -------------------------------------------------------------------------------- /compressai_trainer/utils/git.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/git.py -------------------------------------------------------------------------------- /compressai_trainer/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/metrics.py -------------------------------------------------------------------------------- /compressai_trainer/utils/optimal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/optimal.py -------------------------------------------------------------------------------- /compressai_trainer/utils/pip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/pip.py -------------------------------------------------------------------------------- /compressai_trainer/utils/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/system.py -------------------------------------------------------------------------------- /compressai_trainer/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/compressai_trainer/utils/utils.py -------------------------------------------------------------------------------- /conf/criterion/RateDistortionLoss.yaml: -------------------------------------------------------------------------------- 1 | type: "RateDistortionLoss" 2 | lmbda: null 3 | -------------------------------------------------------------------------------- /conf/dataset/clic2020-mobile/infer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/clic2020-mobile/infer.yaml -------------------------------------------------------------------------------- /conf/dataset/clic2020-mobile/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/clic2020-mobile/train.yaml -------------------------------------------------------------------------------- /conf/dataset/clic2020-mobile/valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/clic2020-mobile/valid.yaml -------------------------------------------------------------------------------- /conf/dataset/clic2020-professional/infer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/clic2020-professional/infer.yaml -------------------------------------------------------------------------------- /conf/dataset/clic2020-professional/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/clic2020-professional/train.yaml -------------------------------------------------------------------------------- /conf/dataset/clic2020-professional/valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/clic2020-professional/valid.yaml -------------------------------------------------------------------------------- /conf/dataset/kodak/infer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/kodak/infer.yaml -------------------------------------------------------------------------------- /conf/dataset/pregenerated/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/pregenerated/train.yaml -------------------------------------------------------------------------------- /conf/dataset/pregenerated/valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/pregenerated/valid.yaml -------------------------------------------------------------------------------- /conf/dataset/vimeo90k/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/vimeo90k/train.yaml -------------------------------------------------------------------------------- /conf/dataset/vimeo90k/valid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/dataset/vimeo90k/valid.yaml -------------------------------------------------------------------------------- /conf/engine/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/engine/default.yaml -------------------------------------------------------------------------------- /conf/env/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/env/default.yaml -------------------------------------------------------------------------------- /conf/eval_zoo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/eval_zoo.yaml -------------------------------------------------------------------------------- /conf/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/example.yaml -------------------------------------------------------------------------------- /conf/example_gvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/example_gvae.yaml -------------------------------------------------------------------------------- /conf/example_gvae_pregenerated.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/example_gvae_pregenerated.yaml -------------------------------------------------------------------------------- /conf/example_pregenerated.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/example_pregenerated.yaml -------------------------------------------------------------------------------- /conf/misc/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/misc/default.yaml -------------------------------------------------------------------------------- /conf/optimizer/net_aux.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/optimizer/net_aux.yaml -------------------------------------------------------------------------------- /conf/paths/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/paths/default.yaml -------------------------------------------------------------------------------- /conf/runner/GVAEImageCompressionRunner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/runner/GVAEImageCompressionRunner.yaml -------------------------------------------------------------------------------- /conf/runner/ImageCompressionRunner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/runner/ImageCompressionRunner.yaml -------------------------------------------------------------------------------- /conf/scheduler/OneCycleLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/scheduler/OneCycleLR.yaml -------------------------------------------------------------------------------- /conf/scheduler/ReduceLROnPlateau.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/conf/scheduler/ReduceLROnPlateau.yaml -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/.gitignore -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/source/_static/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/_static/logo.svg -------------------------------------------------------------------------------- /docs/source/compressai_trainer/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/compressai_trainer/config.rst -------------------------------------------------------------------------------- /docs/source/compressai_trainer/plot.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/compressai_trainer/plot.rst -------------------------------------------------------------------------------- /docs/source/compressai_trainer/registry.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/compressai_trainer/registry.rst -------------------------------------------------------------------------------- /docs/source/compressai_trainer/run.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/compressai_trainer/run.rst -------------------------------------------------------------------------------- /docs/source/compressai_trainer/runners.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/compressai_trainer/runners.rst -------------------------------------------------------------------------------- /docs/source/compressai_trainer/typing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/compressai_trainer/typing.rst -------------------------------------------------------------------------------- /docs/source/compressai_trainer/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/compressai_trainer/utils.rst -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/media/images/aim-run-figure-rd-curves.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/media/images/aim-run-figure-rd-curves.png -------------------------------------------------------------------------------- /docs/source/tools/eval_model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/tools/eval_model.rst -------------------------------------------------------------------------------- /docs/source/tools/rd_plotter.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/tools/rd_plotter.rst -------------------------------------------------------------------------------- /docs/source/tools/train.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/tools/train.rst -------------------------------------------------------------------------------- /docs/source/tutorials/full.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/tutorials/full.rst -------------------------------------------------------------------------------- /docs/source/tutorials/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/docs/source/tutorials/installation.rst -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/SyntheticDataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/tests/SyntheticDataset.yaml -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/tests/test_train.py -------------------------------------------------------------------------------- /tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/tests/utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/InterDigitalInc/CompressAI-Trainer/HEAD/train.py --------------------------------------------------------------------------------