LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
bin/scripts/catalyst-parallel-run
bin/scripts/download-gdrive
bin/scripts/extract-archive
catalyst/__init__.py
catalyst/__version__.py
catalyst/registry.py
catalyst/settings.py
catalyst/typing.py
catalyst/callbacks/__init__.py
catalyst/callbacks/backward.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/sklearn_model.py
catalyst/callbacks/soft_update.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/r2_squared.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/_misc.py
catalyst/contrib/data/collate_fn.py
catalyst/contrib/data/dataset.py
catalyst/contrib/data/dataset_cv.py
catalyst/contrib/data/dataset_ml.py
catalyst/contrib/data/reader.py
catalyst/contrib/data/reader_cv.py
catalyst/contrib/data/sampler.py
catalyst/contrib/data/sampler_inbatch.py
catalyst/contrib/data/transforms.py
catalyst/contrib/datasets/__init__.py
catalyst/contrib/datasets/cifar.py
catalyst/contrib/datasets/imagecar.py
catalyst/contrib/datasets/imagenette.py
catalyst/contrib/datasets/imagewang.py
catalyst/contrib/datasets/imagewoof.py
catalyst/contrib/datasets/market1501.py
catalyst/contrib/datasets/misc.py
catalyst/contrib/datasets/misc_cv.py
catalyst/contrib/datasets/mnist.py
catalyst/contrib/datasets/movielens.py
catalyst/contrib/layers/__init__.py
catalyst/contrib/layers/amsoftmax.py
catalyst/contrib/layers/arcface.py
catalyst/contrib/layers/arcmargin.py
catalyst/contrib/layers/common.py
catalyst/contrib/layers/cosface.py
catalyst/contrib/layers/curricularface.py
catalyst/contrib/layers/factorized.py
catalyst/contrib/layers/lama.py
catalyst/contrib/layers/pooling.py
catalyst/contrib/layers/rms_norm.py
catalyst/contrib/layers/se.py
catalyst/contrib/layers/softmax.py
catalyst/contrib/losses/__init__.py
catalyst/contrib/losses/ce.py
catalyst/contrib/losses/circle.py
catalyst/contrib/losses/contrastive.py
catalyst/contrib/losses/dice.py
catalyst/contrib/losses/focal.py
catalyst/contrib/losses/functional.py
catalyst/contrib/losses/gan.py
catalyst/contrib/losses/iou.py
catalyst/contrib/losses/lovasz.py
catalyst/contrib/losses/margin.py
catalyst/contrib/losses/ntxent.py
catalyst/contrib/losses/recsys.py
catalyst/contrib/losses/regression.py
catalyst/contrib/losses/smoothing_dice.py
catalyst/contrib/losses/supervised_contrastive.py
catalyst/contrib/losses/trevsky.py
catalyst/contrib/losses/triplet.py
catalyst/contrib/losses/wing.py
catalyst/contrib/models/__init__.py
catalyst/contrib/models/mnist.py
catalyst/contrib/models/resnet_encoder.py
catalyst/contrib/optimizers/__init__.py
catalyst/contrib/optimizers/adamp.py
catalyst/contrib/optimizers/lamb.py
catalyst/contrib/optimizers/lookahead.py
catalyst/contrib/optimizers/qhadamw.py
catalyst/contrib/optimizers/radam.py
catalyst/contrib/optimizers/ralamb.py
catalyst/contrib/optimizers/sgdp.py
catalyst/contrib/schedulers/__init__.py
catalyst/contrib/schedulers/base.py
catalyst/contrib/schedulers/onecycle.py
catalyst/contrib/scripts/__init__.py
catalyst/contrib/scripts/collect_env.py
catalyst/contrib/scripts/project_embeddings.py
catalyst/contrib/scripts/run.py
catalyst/contrib/scripts/tune.py
catalyst/contrib/utils/__init__.py
catalyst/contrib/utils/compression.py
catalyst/contrib/utils/image.py
catalyst/contrib/utils/numpy.py
catalyst/contrib/utils/report.py
catalyst/contrib/utils/serialization.py
catalyst/contrib/utils/swa.py
catalyst/contrib/utils/thresholds.py
catalyst/contrib/utils/torch.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/data/__init__.py
catalyst/data/dataset.py
catalyst/data/loader.py
catalyst/data/sampler.py
catalyst/dl/__init__.py
catalyst/engines/__init__.py
catalyst/engines/torch.py
catalyst/extras/__init__.py
catalyst/extras/forward_wrapper.py
catalyst/extras/frozen_class.py
catalyst/extras/metric_handler.py
catalyst/extras/time_manager.py
catalyst/loggers/__init__.py
catalyst/loggers/comet.py
catalyst/loggers/console.py
catalyst/loggers/csv.py
catalyst/loggers/mlflow.py
catalyst/loggers/neptune.py
catalyst/loggers/tensorboard.py
catalyst/loggers/wandb.py
catalyst/metrics/__init__.py
catalyst/metrics/_accumulative.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/_r2_squared.py
catalyst/metrics/_segmentation.py
catalyst/metrics/_topk_metric.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/_r2_squared.py
catalyst/metrics/functional/_recall.py
catalyst/metrics/functional/_segmentation.py
catalyst/runners/__init__.py
catalyst/runners/runner.py
catalyst/runners/supervised.py
catalyst/utils/__init__.py
catalyst/utils/config.py
catalyst/utils/distributed.py
catalyst/utils/misc.py
catalyst/utils/onnx.py
catalyst/utils/pruning.py
catalyst/utils/quantization.py
catalyst/utils/torch.py
catalyst/utils/tracing.py
catalyst_pdm.egg-info/PKG-INFO
catalyst_pdm.egg-info/SOURCES.txt
catalyst_pdm.egg-info/dependency_links.txt
catalyst_pdm.egg-info/entry_points.txt
catalyst_pdm.egg-info/requires.txt
catalyst_pdm.egg-info/top_level.txt
tests/benchmarks/__init__.py
tests/benchmarks/test_benchmark.py
tests/catalyst/__init__.py
tests/catalyst/callbacks/__init__.py
tests/catalyst/callbacks/test_batch_overfit.py
tests/catalyst/callbacks/test_batch_transform.py
tests/catalyst/callbacks/test_checkpoint.py
tests/catalyst/callbacks/test_control_flow.py
tests/catalyst/callbacks/test_metric_aggregation.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_sklearn_model.py
tests/catalyst/callbacks/test_soft_update.py
tests/catalyst/callbacks/test_tracing.py
tests/catalyst/callbacks/test_wrapper.py
tests/catalyst/data/__init__.py
tests/catalyst/data/test_data.py
tests/catalyst/data/test_loader.py
tests/catalyst/data/test_sampler.py
tests/catalyst/dl/__init__.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_r2squared.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_r2_squared.py
tests/catalyst/metrics/functional/test_region_statictics.py
tests/catalyst/runners/__init__.py
tests/catalyst/runners/test_profiler.py
tests/catalyst/runners/test_reid.py
tests/catalyst/runners/test_runner.py
tests/catalyst/runners/test_sklearn_classifier.py
tests/catalyst/runners/test_sklearn_classifier_mnist.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/contrib/__init__.py
tests/contrib/data/__init__.py
tests/contrib/data/test_dataset.py
tests/contrib/data/test_nifti.py
tests/contrib/data/test_sampler_inbatch.py
tests/contrib/datasets/__init__.py
tests/contrib/datasets/test_movielens.py
tests/contrib/datasets/test_movielens_20m.py
tests/contrib/nn/__init__.py
tests/contrib/nn/test_criterion.py
tests/contrib/nn/test_modules.py
tests/contrib/nn/test_onecyle.py
tests/contrib/nn/test_optimizer.py
tests/contrib/scripts/__init__.py
tests/contrib/scripts/test_main.py
tests/contrib/scripts/test_tag2label.py
tests/contrib/scripts/test_tune.py
tests/contrib/utils/__init__.py
tests/contrib/utils/test_image.py
tests/contrib/utils/test_swa.py
tests/contrib/utils/test_torch.py
tests/pipelines/__init__.py
tests/pipelines/test_classification.py
tests/pipelines/test_contrastive.py
tests/pipelines/test_distillation.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