.gitignore
.mergify.yml
.pre-commit-config.yaml
CHANGELOG.md
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
MANIFEST.md
Makefile
README.md
pyproject.toml
setup.cfg
setup.py
.github/CODEOWNERS
.github/FUNDING.yml
.github/PULL_REQUEST_TEMPLATE.md
.github/dependabot.yml
.github/stale.yml
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/documentation.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/ISSUE_TEMPLATE/question.md
.github/workflows/codestyle.yml
.github/workflows/deploy_publish.yml
.github/workflows/deploy_push.yml
.github/workflows/dl_cpu.yml
.github/workflows/dl_cpu_extras.yml
.github/workflows/dl_cpu_minimal.yml
.github/workflows/greetings.yml
.github/workflows/integrations.yml
bin/scripts/catalyst-parallel-run
bin/scripts/download-gdrive
bin/scripts/extract-archive
bin/scripts/install-apex
bin/workflows/check_docs.sh
bin/workflows/check_projector.sh
bin/workflows/check_settings.sh
catalyst/__init__.py
catalyst/__version__.py
catalyst/registry.py
catalyst/settings.py
catalyst/typing.py
catalyst.egg-info/PKG-INFO
catalyst.egg-info/SOURCES.txt
catalyst.egg-info/dependency_links.txt
catalyst.egg-info/entry_points.txt
catalyst.egg-info/requires.txt
catalyst.egg-info/top_level.txt
catalyst/callbacks/__init__.py
catalyst/callbacks/batch_overfit.py
catalyst/callbacks/batch_transform.py
catalyst/callbacks/checkpoint.py
catalyst/callbacks/control_flow.py
catalyst/callbacks/criterion.py
catalyst/callbacks/metric.py
catalyst/callbacks/metric_aggregation.py
catalyst/callbacks/misc.py
catalyst/callbacks/mixup.py
catalyst/callbacks/onnx.py
catalyst/callbacks/optimizer.py
catalyst/callbacks/optuna.py
catalyst/callbacks/periodic_loader.py
catalyst/callbacks/profiler.py
catalyst/callbacks/pruning.py
catalyst/callbacks/quantization.py
catalyst/callbacks/scheduler.py
catalyst/callbacks/tracing.py
catalyst/callbacks/metrics/__init__.py
catalyst/callbacks/metrics/accuracy.py
catalyst/callbacks/metrics/auc.py
catalyst/callbacks/metrics/classification.py
catalyst/callbacks/metrics/cmc_score.py
catalyst/callbacks/metrics/confusion_matrix.py
catalyst/callbacks/metrics/functional_metric.py
catalyst/callbacks/metrics/recsys.py
catalyst/callbacks/metrics/scikit_learn.py
catalyst/callbacks/metrics/segmentation.py
catalyst/contrib/__init__.py
catalyst/contrib/__main__.py
catalyst/contrib/data/__init__.py
catalyst/contrib/data/augmentor.py
catalyst/contrib/data/reader.py
catalyst/contrib/data/cv/__init__.py
catalyst/contrib/data/cv/dataset.py
catalyst/contrib/data/cv/reader.py
catalyst/contrib/data/nifti/__init__.py
catalyst/contrib/data/nifti/reader.py
catalyst/contrib/datasets/README.md
catalyst/contrib/datasets/__init__.py
catalyst/contrib/datasets/cifar.py
catalyst/contrib/datasets/functional.py
catalyst/contrib/datasets/market1501.py
catalyst/contrib/datasets/mnist.py
catalyst/contrib/datasets/movielens.py
catalyst/contrib/datasets/cv/__init__.py
catalyst/contrib/datasets/cv/imagecar.py
catalyst/contrib/datasets/cv/imagenette.py
catalyst/contrib/datasets/cv/imagewang.py
catalyst/contrib/datasets/cv/imagewoof.py
catalyst/contrib/datasets/cv/misc.py
catalyst/contrib/models/__init__.py
catalyst/contrib/models/functional.py
catalyst/contrib/models/hydra.py
catalyst/contrib/models/mnist.py
catalyst/contrib/models/sequential.py
catalyst/contrib/models/cv/__init__.py
catalyst/contrib/models/cv/encoders/__init__.py
catalyst/contrib/models/cv/encoders/resnet.py
catalyst/contrib/models/cv/segmentation/README.md
catalyst/contrib/models/cv/segmentation/__init__.py
catalyst/contrib/models/cv/segmentation/abn.py
catalyst/contrib/models/cv/segmentation/core.py
catalyst/contrib/models/cv/segmentation/fpn.py
catalyst/contrib/models/cv/segmentation/linknet.py
catalyst/contrib/models/cv/segmentation/psp.py
catalyst/contrib/models/cv/segmentation/unet.py
catalyst/contrib/models/cv/segmentation/blocks/__init__.py
catalyst/contrib/models/cv/segmentation/blocks/core.py
catalyst/contrib/models/cv/segmentation/blocks/fpn.py
catalyst/contrib/models/cv/segmentation/blocks/psp.py
catalyst/contrib/models/cv/segmentation/blocks/unet.py
catalyst/contrib/models/cv/segmentation/bridge/__init__.py
catalyst/contrib/models/cv/segmentation/bridge/core.py
catalyst/contrib/models/cv/segmentation/bridge/unet.py
catalyst/contrib/models/cv/segmentation/decoder/__init__.py
catalyst/contrib/models/cv/segmentation/decoder/core.py
catalyst/contrib/models/cv/segmentation/decoder/fpn.py
catalyst/contrib/models/cv/segmentation/decoder/psp.py
catalyst/contrib/models/cv/segmentation/decoder/unet.py
catalyst/contrib/models/cv/segmentation/encoder/__init__.py
catalyst/contrib/models/cv/segmentation/encoder/core.py
catalyst/contrib/models/cv/segmentation/encoder/resnet.py
catalyst/contrib/models/cv/segmentation/encoder/unet.py
catalyst/contrib/models/cv/segmentation/head/__init__.py
catalyst/contrib/models/cv/segmentation/head/core.py
catalyst/contrib/models/cv/segmentation/head/fpn.py
catalyst/contrib/models/cv/segmentation/head/unet.py
catalyst/contrib/nn/__init__.py
catalyst/contrib/nn/criterion/__init__.py
catalyst/contrib/nn/criterion/ce.py
catalyst/contrib/nn/criterion/circle.py
catalyst/contrib/nn/criterion/contrastive.py
catalyst/contrib/nn/criterion/dice.py
catalyst/contrib/nn/criterion/focal.py
catalyst/contrib/nn/criterion/functional.py
catalyst/contrib/nn/criterion/gan.py
catalyst/contrib/nn/criterion/huber.py
catalyst/contrib/nn/criterion/iou.py
catalyst/contrib/nn/criterion/lovasz.py
catalyst/contrib/nn/criterion/margin.py
catalyst/contrib/nn/criterion/trevsky.py
catalyst/contrib/nn/criterion/triplet.py
catalyst/contrib/nn/criterion/wing.py
catalyst/contrib/nn/modules/__init__.py
catalyst/contrib/nn/modules/amsoftmax.py
catalyst/contrib/nn/modules/arcface.py
catalyst/contrib/nn/modules/arcmargin.py
catalyst/contrib/nn/modules/common.py
catalyst/contrib/nn/modules/cosface.py
catalyst/contrib/nn/modules/curricularface.py
catalyst/contrib/nn/modules/factorized.py
catalyst/contrib/nn/modules/lama.py
catalyst/contrib/nn/modules/pooling.py
catalyst/contrib/nn/modules/rms_norm.py
catalyst/contrib/nn/modules/se.py
catalyst/contrib/nn/modules/softmax.py
catalyst/contrib/nn/optimizers/__init__.py
catalyst/contrib/nn/optimizers/adamp.py
catalyst/contrib/nn/optimizers/lamb.py
catalyst/contrib/nn/optimizers/lookahead.py
catalyst/contrib/nn/optimizers/qhadamw.py
catalyst/contrib/nn/optimizers/radam.py
catalyst/contrib/nn/optimizers/ralamb.py
catalyst/contrib/nn/optimizers/sgdp.py
catalyst/contrib/nn/schedulers/__init__.py
catalyst/contrib/nn/schedulers/base.py
catalyst/contrib/nn/schedulers/onecycle.py
catalyst/contrib/scripts/__init__.py
catalyst/contrib/scripts/check_index_model.py
catalyst/contrib/scripts/collect_env.py
catalyst/contrib/scripts/create_index_model.py
catalyst/contrib/scripts/image2embedding.py
catalyst/contrib/scripts/process_images.py
catalyst/contrib/scripts/project_embeddings.py
catalyst/contrib/scripts/split_dataframe.py
catalyst/contrib/scripts/tag2label.py
catalyst/contrib/scripts/text2embedding.py
catalyst/contrib/utils/__init__.py
catalyst/contrib/utils/compression.py
catalyst/contrib/utils/image.py
catalyst/contrib/utils/pandas.py
catalyst/contrib/utils/parallel.py
catalyst/contrib/utils/serialization.py
catalyst/contrib/utils/thresholds.py
catalyst/contrib/utils/visualization.py
catalyst/core/__init__.py
catalyst/core/callback.py
catalyst/core/engine.py
catalyst/core/logger.py
catalyst/core/misc.py
catalyst/core/runner.py
catalyst/core/trial.py
catalyst/data/__init__.py
catalyst/data/collate_fn.py
catalyst/data/loader.py
catalyst/data/sampler.py
catalyst/data/sampler_inbatch.py
catalyst/data/transforms.py
catalyst/data/dataset/__init__.py
catalyst/data/dataset/metric_learning.py
catalyst/data/dataset/torch.py
catalyst/dl/__init__.py
catalyst/dl/__main__.py
catalyst/dl/scripts/__init__.py
catalyst/dl/scripts/hydra_run.py
catalyst/dl/scripts/misc.py
catalyst/dl/scripts/quantize.py
catalyst/dl/scripts/run.py
catalyst/dl/scripts/swa.py
catalyst/dl/scripts/trace.py
catalyst/dl/scripts/tune.py
catalyst/engines/__init__.py
catalyst/engines/amp.py
catalyst/engines/apex.py
catalyst/engines/deepspeed.py
catalyst/engines/fairscale.py
catalyst/engines/torch.py
catalyst/loggers/__init__.py
catalyst/loggers/console.py
catalyst/loggers/csv.py
catalyst/loggers/functional.py
catalyst/loggers/mlflow.py
catalyst/loggers/neptune.py
catalyst/loggers/tensorboard.py
catalyst/loggers/wandb.py
catalyst/metrics/__init__.py
catalyst/metrics/_accuracy.py
catalyst/metrics/_additive.py
catalyst/metrics/_auc.py
catalyst/metrics/_classification.py
catalyst/metrics/_cmc_score.py
catalyst/metrics/_confusion_matrix.py
catalyst/metrics/_functional_metric.py
catalyst/metrics/_hitrate.py
catalyst/metrics/_map.py
catalyst/metrics/_metric.py
catalyst/metrics/_mrr.py
catalyst/metrics/_ndcg.py
catalyst/metrics/_segmentation.py
catalyst/metrics/functional/__init__.py
catalyst/metrics/functional/_accuracy.py
catalyst/metrics/functional/_auc.py
catalyst/metrics/functional/_average_precision.py
catalyst/metrics/functional/_classification.py
catalyst/metrics/functional/_cmc_score.py
catalyst/metrics/functional/_f1_score.py
catalyst/metrics/functional/_focal.py
catalyst/metrics/functional/_hitrate.py
catalyst/metrics/functional/_misc.py
catalyst/metrics/functional/_mrr.py
catalyst/metrics/functional/_ndcg.py
catalyst/metrics/functional/_precision.py
catalyst/metrics/functional/_recall.py
catalyst/metrics/functional/_segmentation.py
catalyst/runners/__init__.py
catalyst/runners/config.py
catalyst/runners/hydra.py
catalyst/runners/misc.py
catalyst/runners/runner.py
catalyst/runners/supervised.py
catalyst/tools/__init__.py
catalyst/tools/forward_wrapper.py
catalyst/tools/frozen_class.py
catalyst/tools/metric_handler.py
catalyst/tools/registry.py
catalyst/tools/time_manager.py
catalyst/utils/__init__.py
catalyst/utils/config.py
catalyst/utils/data.py
catalyst/utils/distributed.py
catalyst/utils/misc.py
catalyst/utils/mixup.py
catalyst/utils/numpy.py
catalyst/utils/onnx.py
catalyst/utils/pruning.py
catalyst/utils/quantization.py
catalyst/utils/swa.py
catalyst/utils/sys.py
catalyst/utils/torch.py
catalyst/utils/tracing.py
docker/Dockerfile
docker/Dockerfile-dev
docker/README.md
docker/collect_dependencies_hash.py
docs/Makefile
docs/README.md
docs/conf.py
docs/create_api.sh
docs/index.rst
docs/key.enc
docs/requirements.txt
docs/api/callbacks.rst
docs/api/contrib.rst
docs/api/core.rst
docs/api/data.rst
docs/api/engines.rst
docs/api/loggers.rst
docs/api/metrics.rst
docs/api/runners.rst
docs/api/tools.rst
docs/api/utils.rst
docs/core/callback.rst
docs/core/engine.rst
docs/core/logger.rst
docs/core/metric.rst
docs/core/runner.rst
docs/faq/amp.rst
docs/faq/checkpointing.rst
docs/faq/config_api.rst
docs/faq/data.rst
docs/faq/ddp.rst
docs/faq/debugging.rst
docs/faq/dp.rst
docs/faq/early_stopping.rst
docs/faq/engines.rst
docs/faq/finetuning.rst
docs/faq/inference.rst
docs/faq/intro.rst
docs/faq/logging.rst
docs/faq/multi_components.rst
docs/faq/multiple_keys.rst
docs/faq/optuna.rst
docs/getting_started/migrating_from_other.rst
docs/getting_started/quickstart.rst
docs/tutorials/ddp.rst
examples/README.md
examples/catalyst_rl/README.md
examples/catalyst_rl/buffer.py
examples/catalyst_rl/db.py
examples/catalyst_rl/ddpg.py
examples/catalyst_rl/dqn.py
examples/catalyst_rl/env.py
examples/catalyst_rl/misc.py
examples/catalyst_rl/sampler.py
examples/configs/config-description-eng.yml
examples/configs/config-description-rus.yml
examples/engines/README.md
examples/engines/multi_gpu.py
examples/mnist_stages/README.md
examples/mnist_stages/__init__.py
examples/mnist_stages/config.yml
examples/mnist_stages/config_hydra.yaml
examples/mnist_stages/config_tune.yml
examples/mnist_stages/model.py
examples/mnist_stages/run_config.sh
examples/mnist_stages/run_hydra.sh
examples/mnist_stages/run_tune.sh
examples/mnist_stages/runner.py
examples/notebooks/colab_ci_cd.ipynb
examples/notebooks/customization_tutorial.ipynb
examples/notebooks/customizing_what_happens_in_train.ipynb
examples/notebooks/reinforcement_learning.ipynb
examples/reinforcement_learning/README.md
examples/reinforcement_learning/ddpg.py
examples/reinforcement_learning/dqn.py
examples/reinforcement_learning/reinforce.py
requirements/requirements-albu.txt
requirements/requirements-cv.txt
requirements/requirements-deepspeed.txt
requirements/requirements-dev.txt
requirements/requirements-fairscale.txt
requirements/requirements-hydra.txt
requirements/requirements-ml.txt
requirements/requirements-mlflow.txt
requirements/requirements-neptune.txt
requirements/requirements-nifti.txt
requirements/requirements-onnx-gpu.txt
requirements/requirements-onnx.txt
requirements/requirements-optuna.txt
requirements/requirements-wandb.txt
requirements/requirements.txt
tests/__init__.py
tests/catalyst/__init__.py
tests/catalyst/callbacks/__init__.py
tests/catalyst/callbacks/test_aggregation.py
tests/catalyst/callbacks/test_batch_overfit.py
tests/catalyst/callbacks/test_checkpoint.py
tests/catalyst/callbacks/test_control_flow.py
tests/catalyst/callbacks/test_metric.py
tests/catalyst/callbacks/test_misc.py
tests/catalyst/callbacks/test_mixup.py
tests/catalyst/callbacks/test_optimizer.py
tests/catalyst/callbacks/test_optuna.py
tests/catalyst/callbacks/test_periodic_loader.py
tests/catalyst/callbacks/test_profiler.py
tests/catalyst/callbacks/test_pruning.py
tests/catalyst/callbacks/test_quantization.py
tests/catalyst/callbacks/test_tracing.py
tests/catalyst/callbacks/test_transform_kornia.py
tests/catalyst/callbacks/test_wrapper_callback.py
tests/catalyst/contrib/__init__.py
tests/catalyst/contrib/data/__init__.py
tests/catalyst/contrib/data/test_nifti.py
tests/catalyst/contrib/datasets/__init__.py
tests/catalyst/contrib/datasets/test_movielens.py
tests/catalyst/contrib/models/__init__.py
tests/catalyst/contrib/models/config3.yml
tests/catalyst/contrib/models/config4.yml
tests/catalyst/contrib/models/test_functional.py
tests/catalyst/contrib/models/test_hydra.py
tests/catalyst/contrib/models/test_segmentation.py
tests/catalyst/contrib/nn/__init__.py
tests/catalyst/contrib/nn/test_criterion.py
tests/catalyst/contrib/nn/test_modules.py
tests/catalyst/contrib/nn/test_onecyle.py
tests/catalyst/contrib/nn/test_optimizer.py
tests/catalyst/contrib/scipts/__init__.py
tests/catalyst/contrib/scipts/test_tag2label.py
tests/catalyst/contrib/utils/__init__.py
tests/catalyst/contrib/utils/test_image.py
tests/catalyst/contrib/utils/test_misc.py
tests/catalyst/contrib/utils/test_pandas.py
tests/catalyst/data/__init__.py
tests/catalyst/data/test_dataset.py
tests/catalyst/data/test_loader.py
tests/catalyst/data/test_sampler.py
tests/catalyst/data/test_sampler_inbatch.py
tests/catalyst/dl/__init__.py
tests/catalyst/dl/test_main.py
tests/catalyst/dl/test_runner.py
tests/catalyst/engines/__init__.py
tests/catalyst/engines/misc.py
tests/catalyst/engines/test_amp.py
tests/catalyst/engines/test_apex.py
tests/catalyst/engines/test_device.py
tests/catalyst/engines/test_distributed.py
tests/catalyst/engines/test_distributed_amp.py
tests/catalyst/engines/test_distributed_apex.py
tests/catalyst/engines/test_functional.py
tests/catalyst/engines/test_metrics.py
tests/catalyst/engines/test_parallel.py
tests/catalyst/engines/test_parallel_amp.py
tests/catalyst/engines/test_parallel_apex.py
tests/catalyst/metrics/__init__.py
tests/catalyst/metrics/test_accuracy.py
tests/catalyst/metrics/test_additive.py
tests/catalyst/metrics/test_classification.py
tests/catalyst/metrics/test_cmc_score.py
tests/catalyst/metrics/test_functional_metric.py
tests/catalyst/metrics/test_segmentation.py
tests/catalyst/metrics/functional/__init__.py
tests/catalyst/metrics/functional/test_accuracy.py
tests/catalyst/metrics/functional/test_auc.py
tests/catalyst/metrics/functional/test_average_precision.py
tests/catalyst/metrics/functional/test_classification.py
tests/catalyst/metrics/functional/test_cmc_metric.py
tests/catalyst/metrics/functional/test_dice.py
tests/catalyst/metrics/functional/test_fbeta_precision_recall.py
tests/catalyst/metrics/functional/test_hitrate.py
tests/catalyst/metrics/functional/test_iou.py
tests/catalyst/metrics/functional/test_misc.py
tests/catalyst/metrics/functional/test_mrr.py
tests/catalyst/metrics/functional/test_ndcg.py
tests/catalyst/metrics/functional/test_region_statictics.py
tests/catalyst/runners/__init__.py
tests/catalyst/runners/test_runner.py
tests/catalyst/runners/test_train_flags.py
tests/catalyst/tools/__init__.py
tests/catalyst/tools/registery_foo.py
tests/catalyst/tools/test_registry.py
tests/catalyst/utils/__init__.py
tests/catalyst/utils/test_config.py
tests/catalyst/utils/test_hash.py
tests/catalyst/utils/test_onnx.py
tests/catalyst/utils/test_quantization.py
tests/catalyst/utils/test_swa.py
tests/catalyst/utils/test_torch.py
tests/pipelines/__init__.py
tests/pipelines/test_classification.py
tests/pipelines/test_config_runner.py
tests/pipelines/test_data.py
tests/pipelines/test_distillation.py
tests/pipelines/test_finetune.py
tests/pipelines/test_finetune2.py
tests/pipelines/test_gan.py
tests/pipelines/test_metric_learning.py
tests/pipelines/test_mnist.py
tests/pipelines/test_mnist_custom.py
tests/pipelines/test_mnist_multicriterion.py
tests/pipelines/test_mnist_multimodel.py
tests/pipelines/test_mnist_multioptimizer.py
tests/pipelines/test_multihead_classification.py
tests/pipelines/test_multilabel_classification.py
tests/pipelines/test_optuna.py
tests/pipelines/test_recsys.py
tests/pipelines/test_regression.py
tests/pipelines/test_vae.py