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.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/contrib/__init__.py
catalyst/contrib/__main__.py
catalyst/contrib/registry.py
catalyst/contrib/data/__init__.py
catalyst/contrib/datasets/__init__.py
catalyst/contrib/datasets/functional.py
catalyst/contrib/datasets/mnist.py
catalyst/contrib/datasets/cv/__init__.py
catalyst/contrib/datasets/cv/fastai.py
catalyst/contrib/datasets/cv/imagenette.py
catalyst/contrib/datasets/cv/imagewang.py
catalyst/contrib/datasets/cv/imagewoof.py
catalyst/contrib/dl/__init__.py
catalyst/contrib/dl/callbacks/__init__.py
catalyst/contrib/dl/callbacks/alchemy_logger.py
catalyst/contrib/dl/callbacks/cutmix_callback.py
catalyst/contrib/dl/callbacks/gradnorm_logger.py
catalyst/contrib/dl/callbacks/knn_metric.py
catalyst/contrib/dl/callbacks/kornia_transform.py
catalyst/contrib/dl/callbacks/mask_inference.py
catalyst/contrib/dl/callbacks/neptune_logger.py
catalyst/contrib/dl/callbacks/optuna_callback.py
catalyst/contrib/dl/callbacks/perplexity_metric.py
catalyst/contrib/dl/callbacks/telegram_logger.py
catalyst/contrib/dl/callbacks/visdom_logger.py
catalyst/contrib/dl/callbacks/wandb_logger.py
catalyst/contrib/dl/callbacks/tests/__init__.py
catalyst/contrib/dl/callbacks/tests/test_gradnorm_logger.py
catalyst/contrib/dl/callbacks/tests/test_optuna_callback.py
catalyst/contrib/dl/callbacks/tests/test_perplexity_callback.py
catalyst/contrib/dl/callbacks/tests/test_tracer_callback.py
catalyst/contrib/dl/experiment/__init__.py
catalyst/contrib/dl/runner/__init__.py
catalyst/contrib/models/__init__.py
catalyst/contrib/models/functional.py
catalyst/contrib/models/hydra.py
catalyst/contrib/models/sequential.py
catalyst/contrib/models/simple_conv.py
catalyst/contrib/models/cv/__init__.py
catalyst/contrib/models/cv/classification/__init__.py
catalyst/contrib/models/cv/encoders/__init__.py
catalyst/contrib/models/cv/encoders/resnet.py
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/models/nlp/__init__.py
catalyst/contrib/models/nlp/classification/__init__.py
catalyst/contrib/models/nlp/classification/bert.py
catalyst/contrib/models/nlp/encoders/__init__.py
catalyst/contrib/models/tests/__init__.py
catalyst/contrib/models/tests/test_hydra.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/triplet.py
catalyst/contrib/nn/criterion/wing.py
catalyst/contrib/nn/modules/__init__.py
catalyst/contrib/nn/modules/common.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/optimizers/__init__.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/schedulers/__init__.py
catalyst/contrib/nn/schedulers/base.py
catalyst/contrib/nn/schedulers/onecycle.py
catalyst/contrib/nn/tests/__init__.py
catalyst/contrib/nn/tests/test_criterion.py
catalyst/contrib/nn/tests/test_optimizer.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/find_thresholds.py
catalyst/contrib/tools/__init__.py
catalyst/contrib/tools/tensorboard.py
catalyst/contrib/tools/tests/__init__.py
catalyst/contrib/tools/tests/test_tensorboard.py
catalyst/contrib/utils/__init__.py
catalyst/contrib/utils/argparse.py
catalyst/contrib/utils/compression.py
catalyst/contrib/utils/confusion_matrix.py
catalyst/contrib/utils/dataset.py
catalyst/contrib/utils/misc.py
catalyst/contrib/utils/pandas.py
catalyst/contrib/utils/parallel.py
catalyst/contrib/utils/plotly.py
catalyst/contrib/utils/serialization.py
catalyst/contrib/utils/visualization.py
catalyst/contrib/utils/cv/__init__.py
catalyst/contrib/utils/cv/image.py
catalyst/contrib/utils/cv/tensor.py
catalyst/contrib/utils/cv/tests/__init__.py
catalyst/contrib/utils/cv/tests/test_image.py
catalyst/contrib/utils/nlp/__init__.py
catalyst/contrib/utils/nlp/text.py
catalyst/contrib/utils/tests/__init__.py
catalyst/contrib/utils/tests/test_argparse.py
catalyst/contrib/utils/tests/test_dataset.py
catalyst/contrib/utils/tests/test_misc.py
catalyst/contrib/utils/tests/test_pandas.py
catalyst/core/__init__.py
catalyst/core/callback.py
catalyst/core/experiment.py
catalyst/core/legacy.py
catalyst/core/registry.py
catalyst/core/runner.py
catalyst/core/state.py
catalyst/core/callbacks/__init__.py
catalyst/core/callbacks/batch_overfit.py
catalyst/core/callbacks/checkpoint.py
catalyst/core/callbacks/control_flow.py
catalyst/core/callbacks/criterion.py
catalyst/core/callbacks/early_stop.py
catalyst/core/callbacks/exception.py
catalyst/core/callbacks/formatters.py
catalyst/core/callbacks/logging.py
catalyst/core/callbacks/metrics.py
catalyst/core/callbacks/optimizer.py
catalyst/core/callbacks/periodic_loader.py
catalyst/core/callbacks/scheduler.py
catalyst/core/callbacks/timer.py
catalyst/core/callbacks/validation.py
catalyst/core/callbacks/tests/__init__.py
catalyst/core/callbacks/tests/test_checkpoint_callback.py
catalyst/core/callbacks/tests/test_control_flow_callback.py
catalyst/core/callbacks/tests/test_early_stop.py
catalyst/core/callbacks/tests/test_periodic_loader_callback.py
catalyst/core/callbacks/tests/test_pruning.py
catalyst/core/tests/__init__.py
catalyst/core/tests/test_optimizer_callback.py
catalyst/core/tests/test_wrapper_callback.py
catalyst/core/utils/__init__.py
catalyst/core/utils/callbacks.py
catalyst/core/utils/data.py
catalyst/data/__init__.py
catalyst/data/__main__.py
catalyst/data/augmentor.py
catalyst/data/collate_fn.py
catalyst/data/loader.py
catalyst/data/reader.py
catalyst/data/sampler.py
catalyst/data/sampler_inbatch.py
catalyst/data/utils.py
catalyst/data/cv/__init__.py
catalyst/data/cv/dataset.py
catalyst/data/cv/reader.py
catalyst/data/cv/mixins/__init__.py
catalyst/data/cv/mixins/blur.py
catalyst/data/cv/mixins/flare.py
catalyst/data/cv/mixins/rotate.py
catalyst/data/cv/mixins/tests/__init__.py
catalyst/data/cv/mixins/tests/test_mixin.py
catalyst/data/cv/transforms/__init__.py
catalyst/data/cv/transforms/albumentations.py
catalyst/data/cv/transforms/torch.py
catalyst/data/dataset/__init__.py
catalyst/data/dataset/metric_learning.py
catalyst/data/dataset/torch.py
catalyst/data/nlp/__init__.py
catalyst/data/nlp/dataset/__init__.py
catalyst/data/nlp/dataset/language_modeling.py
catalyst/data/nlp/dataset/text_classification.py
catalyst/data/nlp/dataset/tests/__init__.py
catalyst/data/nlp/dataset/tests/test_language_modeling_dataset.py
catalyst/data/nlp/dataset/tests/test_text_classification_dataset.py
catalyst/data/scripts/__init__.py
catalyst/data/scripts/image2embedding.py
catalyst/data/scripts/process_images.py
catalyst/data/scripts/project_embeddings.py
catalyst/data/scripts/split_dataframe.py
catalyst/data/scripts/tag2label.py
catalyst/data/scripts/text2embedding.py
catalyst/data/scripts/tests/__init__.py
catalyst/data/scripts/tests/test_tag2label.py
catalyst/data/tests/__init__.py
catalyst/data/tests/test_dataset.py
catalyst/data/tests/test_sampler.py
catalyst/data/tests/test_sampler_inbatch.py
catalyst/dl/__init__.py
catalyst/dl/__main__.py
catalyst/dl/registry.py
catalyst/dl/callbacks/__init__.py
catalyst/dl/callbacks/confusion_matrix.py
catalyst/dl/callbacks/inference.py
catalyst/dl/callbacks/meter.py
catalyst/dl/callbacks/mixup.py
catalyst/dl/callbacks/pruning.py
catalyst/dl/callbacks/quantization.py
catalyst/dl/callbacks/scheduler.py
catalyst/dl/callbacks/tracing.py
catalyst/dl/callbacks/metrics/__init__.py
catalyst/dl/callbacks/metrics/accuracy.py
catalyst/dl/callbacks/metrics/auc.py
catalyst/dl/callbacks/metrics/cmc_score.py
catalyst/dl/callbacks/metrics/dice.py
catalyst/dl/callbacks/metrics/f1_score.py
catalyst/dl/callbacks/metrics/iou.py
catalyst/dl/callbacks/metrics/ppv_tpr_f1.py
catalyst/dl/callbacks/metrics/precision.py
catalyst/dl/callbacks/tests/__init__.py
catalyst/dl/callbacks/tests/test_quantization.py
catalyst/dl/experiment/__init__.py
catalyst/dl/experiment/config.py
catalyst/dl/experiment/experiment.py
catalyst/dl/experiment/supervised.py
catalyst/dl/runner/__init__.py
catalyst/dl/runner/runner.py
catalyst/dl/runner/supervised.py
catalyst/dl/scripts/__init__.py
catalyst/dl/scripts/init.py
catalyst/dl/scripts/quantize.py
catalyst/dl/scripts/run.py
catalyst/dl/scripts/trace.py
catalyst/dl/tests/__init__.py
catalyst/dl/tests/test_main.py
catalyst/dl/tests/test_runner.py
catalyst/dl/utils/__init__.py
catalyst/dl/utils/callbacks.py
catalyst/dl/utils/quantization.py
catalyst/dl/utils/torch.py
catalyst/dl/utils/trace.py
catalyst/dl/utils/wizard.py
catalyst/metrics/__init__.py
catalyst/metrics/functional.py
catalyst/registry/__init__.py
catalyst/registry/registries.py
catalyst/registry/registry.py
catalyst/registry/tests/__init__.py
catalyst/registry/tests/registery_foo.py
catalyst/registry/tests/test_registry.py
catalyst/tools/__init__.py
catalyst/tools/frozen_class.py
catalyst/tools/settings.py
catalyst/tools/time_manager.py
catalyst/tools/typing.py
catalyst/tools/meters/__init__.py
catalyst/tools/meters/apmeter.py
catalyst/tools/meters/aucmeter.py
catalyst/tools/meters/averagevaluemeter.py
catalyst/tools/meters/classerrormeter.py
catalyst/tools/meters/confusionmeter.py
catalyst/tools/meters/mapmeter.py
catalyst/tools/meters/meter.py
catalyst/tools/meters/movingaveragevaluemeter.py
catalyst/tools/meters/msemeter.py
catalyst/tools/meters/ppv_tpr_f1_meter.py
catalyst/tools/meters/tests/__init__.py
catalyst/tools/meters/tests/test_averagevaluemeter.py
catalyst/tools/meters/tests/test_ppv_tpr_f1.py
catalyst/utils/__init__.py
catalyst/utils/checkpoint.py
catalyst/utils/components.py
catalyst/utils/config.py
catalyst/utils/dict.py
catalyst/utils/distributed.py
catalyst/utils/hash.py
catalyst/utils/initialization.py
catalyst/utils/loader.py
catalyst/utils/misc.py
catalyst/utils/numpy.py
catalyst/utils/parser.py
catalyst/utils/pipelines.py
catalyst/utils/pruning.py
catalyst/utils/scripts.py
catalyst/utils/seed.py
catalyst/utils/sys.py
catalyst/utils/torch.py
catalyst/utils/metrics/__init__.py
catalyst/utils/metrics/accuracy.py
catalyst/utils/metrics/auc.py
catalyst/utils/metrics/cmc_score.py
catalyst/utils/metrics/dice.py
catalyst/utils/metrics/f1_score.py
catalyst/utils/metrics/focal.py
catalyst/utils/metrics/functional.py
catalyst/utils/metrics/iou.py
catalyst/utils/metrics/precision.py
catalyst/utils/metrics/tests/__init__.py
catalyst/utils/metrics/tests/test_accuracy.py
catalyst/utils/metrics/tests/test_auc.py
catalyst/utils/metrics/tests/test_average_precision.py
catalyst/utils/metrics/tests/test_cmc_metric.py
catalyst/utils/metrics/tests/test_dice.py
catalyst/utils/metrics/tests/test_iou.py
catalyst/utils/tests/__init__.py
catalyst/utils/tests/test_config.py
catalyst/utils/tests/test_hash.py
catalyst/utils/tests/test_torch.py
tests/_tests_contrib_dl_callbacks/__init__.py
tests/_tests_contrib_dl_callbacks/experiment.py
tests/_tests_contrib_dl_callbacks/model.py
tests/_tests_cv_classification/__init__.py
tests/_tests_cv_classification/experiment.py
tests/_tests_cv_classification/model.py
tests/_tests_cv_classification_transforms/__init__.py
tests/_tests_cv_classification_transforms/experiment.py
tests/_tests_cv_classification_transforms/model.py
tests/_tests_cv_segmentation/__init__.py
tests/_tests_cv_segmentation/dataset.py
tests/_tests_cv_segmentation/experiment.py
tests/_tests_dl_callbacks/__init__.py
tests/_tests_dl_callbacks/experiment.py
tests/_tests_dl_callbacks/model.py
tests/_tests_ml_cmcscore/__init__.py
tests/_tests_ml_cmcscore/experiment.py
tests/_tests_ml_cmcscore/net.py
tests/_tests_scripts/__init__.py
tests/_tests_scripts/core.py
tests/_tests_scripts/core_batch_overfit_callback.py
tests/_tests_scripts/core_contrib_data.py
tests/_tests_scripts/core_contrib_dl.py
tests/_tests_scripts/core_contrib_models.py
tests/_tests_scripts/core_contrib_nn.py
tests/_tests_scripts/core_data.py
tests/_tests_scripts/core_runner.py
tests/_tests_scripts/core_runner_onecyle_lr_scheduler.py
tests/_tests_scripts/cv_contrib_models.py
tests/_tests_scripts/cv_z_kornia.py
tests/_tests_scripts/cv_z_segmentation.py
tests/_tests_scripts/cv_z_unets.py
tests/_tests_scripts/dl_callbacks.py
tests/_tests_scripts/dl_experiment.py
tests/_tests_scripts/dl_runner.py
tests/_tests_scripts/dl_utils.py
tests/_tests_scripts/dl_z_classification.py
tests/_tests_scripts/dl_z_contirb_functional.py
tests/_tests_scripts/dl_z_distributed_01.py
tests/_tests_scripts/dl_z_distributed_02.py
tests/_tests_scripts/dl_z_distributed_03.py
tests/_tests_scripts/dl_z_distributed_04.py
tests/_tests_scripts/dl_z_distributed_05.py
tests/_tests_scripts/dl_z_distributed_06.py
tests/_tests_scripts/dl_z_distributed_07.py
tests/_tests_scripts/dl_z_distributed_08.py
tests/_tests_scripts/dl_z_distributed_09.py
tests/_tests_scripts/dl_z_distributed_10.py
tests/_tests_scripts/dl_z_distributed_11.py
tests/_tests_scripts/dl_z_distributed_12.py
tests/_tests_scripts/dl_z_distributed_13.py
tests/_tests_scripts/dl_z_distributed_14.py
tests/_tests_scripts/dl_z_distributed_15.py
tests/_tests_scripts/dl_z_distributed_16.py
tests/_tests_scripts/dl_z_distributed_17.py
tests/_tests_scripts/dl_z_docs_distributed_0.py
tests/_tests_scripts/dl_z_docs_distributed_1.py
tests/_tests_scripts/dl_z_docs_distributed_2.py
tests/_tests_scripts/dl_z_docs_distributed_3.py
tests/_tests_scripts/dl_z_mvp_classification.py
tests/_tests_scripts/dl_z_mvp_classification2.py
tests/_tests_scripts/dl_z_mvp_distributed_mnist_ae.py
tests/_tests_scripts/dl_z_mvp_mnist.py
tests/_tests_scripts/dl_z_mvp_mnist_ae.py
tests/_tests_scripts/dl_z_mvp_mnist_gan.py
tests/_tests_scripts/dl_z_mvp_mnist_metric_learning.py
tests/_tests_scripts/dl_z_mvp_mnist_unet.py
tests/_tests_scripts/dl_z_mvp_mnist_vae.py
tests/_tests_scripts/dl_z_mvp_regression.py
tests/_tests_scripts/nlp_contrib_models.py