README.md
setup.cfg
setup.py
bin/scripts/catalyst-parallel-run
bin/scripts/catalyst-rl-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/data/cv/__init__.py
catalyst/contrib/data/cv/mixins/__init__.py
catalyst/contrib/data/cv/mixins/blur.py
catalyst/contrib/data/cv/mixins/flare.py
catalyst/contrib/data/cv/mixins/rotate.py
catalyst/contrib/data/cv/mixins/tests/__init__.py
catalyst/contrib/data/cv/mixins/tests/test_mixin.py
catalyst/contrib/data/cv/transforms/__init__.py
catalyst/contrib/data/cv/transforms/tensor.py
catalyst/contrib/data/nlp/__init__.py
catalyst/contrib/data/nlp/classification/__init__.py
catalyst/contrib/data/nlp/classification/dataset.py
catalyst/contrib/data/nlp/classification/tests/__init__.py
catalyst/contrib/data/nlp/classification/tests/test_text_clf_dataset.py
catalyst/contrib/dl/__init__.py
catalyst/contrib/dl/callbacks/__init__.py
catalyst/contrib/dl/callbacks/alchemy.py
catalyst/contrib/dl/callbacks/criterion.py
catalyst/contrib/dl/callbacks/cutmix_callback.py
catalyst/contrib/dl/callbacks/knn.py
catalyst/contrib/dl/callbacks/neptune.py
catalyst/contrib/dl/callbacks/telegram_logger.py
catalyst/contrib/dl/experiment/__init__.py
catalyst/contrib/dl/runner/__init__.py
catalyst/contrib/dl/runner/alchemy.py
catalyst/contrib/dl/runner/neptune.py
catalyst/contrib/dl/runner/wandb.py
catalyst/contrib/models/__init__.py
catalyst/contrib/models/functional.py
catalyst/contrib/models/sequential.py
catalyst/contrib/models/cv/__init__.py
catalyst/contrib/models/cv/classification/__init__.py
catalyst/contrib/models/cv/classification/mobilenetv2.py
catalyst/contrib/models/cv/classification/mobilenetv3.py
catalyst/contrib/models/cv/encoders/__init__.py
catalyst/contrib/models/cv/encoders/_mobilenetv2.py
catalyst/contrib/models/cv/encoders/mobilenetv2.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/cv/segmentation/models/__init__.py
catalyst/contrib/models/cv/segmentation/models/mobileunet.py
catalyst/contrib/models/cv/segmentation/models/resnetlinknet.py
catalyst/contrib/models/cv/segmentation/models/resnetunet.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/nlp/segmentation/__init__.py
catalyst/contrib/nn/__init__.py
catalyst/contrib/nn/criterion/__init__.py
catalyst/contrib/nn/criterion/ce.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/noisy.py
catalyst/contrib/nn/modules/pooling.py
catalyst/contrib/nn/modules/real_nvp.py
catalyst/contrib/nn/modules/rms_norm.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/rl/__init__.py
catalyst/contrib/scripts/__init__.py
catalyst/contrib/scripts/check_index_model.py
catalyst/contrib/scripts/create_index_model.py
catalyst/contrib/scripts/find_thresholds.py
catalyst/core/__init__.py
catalyst/core/callback.py
catalyst/core/experiment.py
catalyst/core/registry.py
catalyst/core/runner.py
catalyst/core/state.py
catalyst/core/callbacks/__init__.py
catalyst/core/callbacks/checkpoint.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/phase.py
catalyst/core/callbacks/scheduler.py
catalyst/core/callbacks/timer.py
catalyst/core/callbacks/validation.py
catalyst/core/callbacks/wrappers.py
catalyst/data/__init__.py
catalyst/data/__main__.py
catalyst/data/augmentor.py
catalyst/data/collate_fn.py
catalyst/data/dataset.py
catalyst/data/reader.py
catalyst/data/sampler.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/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/gan.py
catalyst/dl/callbacks/inference.py
catalyst/dl/callbacks/mixup.py
catalyst/dl/callbacks/scheduler.py
catalyst/dl/callbacks/metrics/__init__.py
catalyst/dl/callbacks/metrics/accuracy.py
catalyst/dl/callbacks/metrics/auc.py
catalyst/dl/callbacks/metrics/dice.py
catalyst/dl/callbacks/metrics/f1_score.py
catalyst/dl/callbacks/metrics/functional.py
catalyst/dl/callbacks/metrics/iou.py
catalyst/dl/callbacks/metrics/ppv_tpr_f1.py
catalyst/dl/core/__init__.py
catalyst/dl/core/callback.py
catalyst/dl/core/experiment.py
catalyst/dl/core/runner.py
catalyst/dl/core/state.py
catalyst/dl/experiment/__init__.py
catalyst/dl/experiment/base.py
catalyst/dl/experiment/config.py
catalyst/dl/experiment/gan.py
catalyst/dl/experiment/supervised.py
catalyst/dl/runner/__init__.py
catalyst/dl/runner/gan.py
catalyst/dl/runner/supervised.py
catalyst/dl/scripts/__init__.py
catalyst/dl/scripts/init.py
catalyst/dl/scripts/run.py
catalyst/dl/scripts/trace.py
catalyst/dl/state/__init__.py
catalyst/dl/state/gan.py
catalyst/dl/tests/__init__.py
catalyst/dl/tests/test_dataset.py
catalyst/dl/tests/test_main.py
catalyst/dl/utils/__init__.py
catalyst/dl/utils/pipelines.py
catalyst/dl/utils/text.py
catalyst/dl/utils/torch.py
catalyst/dl/utils/trace.py
catalyst/dl/utils/visualization.py
catalyst/dl/utils/wizard.py
catalyst/rl/__init__.py
catalyst/rl/__main__.py
catalyst/rl/registry.py
catalyst/rl/agent/__init__.py
catalyst/rl/agent/actor.py
catalyst/rl/agent/critic.py
catalyst/rl/agent/head.py
catalyst/rl/agent/network.py
catalyst/rl/agent/policy.py
catalyst/rl/core/__init__.py
catalyst/rl/core/agent.py
catalyst/rl/core/algorithm.py
catalyst/rl/core/db.py
catalyst/rl/core/environment.py
catalyst/rl/core/exploration.py
catalyst/rl/core/policy_handler.py
catalyst/rl/core/sampler.py
catalyst/rl/core/trainer.py
catalyst/rl/core/trajectory_sampler.py
catalyst/rl/db/__init__.py
catalyst/rl/db/mongo.py
catalyst/rl/db/redis.py
catalyst/rl/environment/__init__.py
catalyst/rl/environment/atari.py
catalyst/rl/environment/env_wrappers.py
catalyst/rl/environment/environment.py
catalyst/rl/environment/gym.py
catalyst/rl/exploration/__init__.py
catalyst/rl/exploration/boltzman.py
catalyst/rl/exploration/gauss.py
catalyst/rl/exploration/greedy.py
catalyst/rl/exploration/param_noise.py
catalyst/rl/offpolicy/__init__.py
catalyst/rl/offpolicy/trainer.py
catalyst/rl/offpolicy/algorithms/__init__.py
catalyst/rl/offpolicy/algorithms/actor_critic.py
catalyst/rl/offpolicy/algorithms/critic.py
catalyst/rl/offpolicy/algorithms/ddpg.py
catalyst/rl/offpolicy/algorithms/dqn.py
catalyst/rl/offpolicy/algorithms/sac.py
catalyst/rl/offpolicy/algorithms/td3.py
catalyst/rl/onpolicy/__init__.py
catalyst/rl/onpolicy/trainer.py
catalyst/rl/onpolicy/algorithms/__init__.py
catalyst/rl/onpolicy/algorithms/actor.py
catalyst/rl/onpolicy/algorithms/actor_critic.py
catalyst/rl/onpolicy/algorithms/ppo.py
catalyst/rl/onpolicy/algorithms/reinforce.py
catalyst/rl/scripts/__init__.py
catalyst/rl/scripts/dump_db.py
catalyst/rl/scripts/load_db.py
catalyst/rl/scripts/misc.py
catalyst/rl/scripts/run_samplers.py
catalyst/rl/scripts/run_trainer.py
catalyst/rl/utils/__init__.py
catalyst/rl/utils/agent.py
catalyst/rl/utils/buffer.py
catalyst/rl/utils/gamma.py
catalyst/rl/utils/gym.py
catalyst/rl/utils/sampler.py
catalyst/rl/utils/torch.py
catalyst/rl/utils/trajectory.py
catalyst/rl/utils/criterion/__init__.py
catalyst/rl/utils/criterion/categorical.py
catalyst/rl/utils/criterion/quantile.py
catalyst/rl2/__init__.py
catalyst/rl2/__main__.py
catalyst/rl2/registry.py
catalyst/rl2/agent/__init__.py
catalyst/rl2/algorithm/__init__.py
catalyst/rl2/callbacks/__init__.py
catalyst/rl2/core/__init__.py
catalyst/rl2/core/experiment.py
catalyst/rl2/core/runner.py
catalyst/rl2/core/state.py
catalyst/rl2/db/__init__.py
catalyst/rl2/environment/__init__.py
catalyst/rl2/experiment/__init__.py
catalyst/rl2/exploration/__init__.py
catalyst/rl2/runner/__init__.py
catalyst/rl2/runner/offpolicy.py
catalyst/rl2/scripts/__init__.py
catalyst/rl2/tests/__init__.py
catalyst/rl2/utils/__init__.py
catalyst/rl2/worker/__init__.py
catalyst/utils/__init__.py
catalyst/utils/argparse.py
catalyst/utils/callbacks.py
catalyst/utils/checkpoint.py
catalyst/utils/compression.py
catalyst/utils/config.py
catalyst/utils/confusion_matrix.py
catalyst/utils/dataset.py
catalyst/utils/ddp.py
catalyst/utils/dict.py
catalyst/utils/distributed.py
catalyst/utils/hash.py
catalyst/utils/image.py
catalyst/utils/initialization.py
catalyst/utils/misc.py
catalyst/utils/notebook.py
catalyst/utils/numpy.py
catalyst/utils/pandas.py
catalyst/utils/parallel.py
catalyst/utils/parser.py
catalyst/utils/plotly.py
catalyst/utils/scripts.py
catalyst/utils/seed.py
catalyst/utils/serialization.py
catalyst/utils/sys.py
catalyst/utils/torch.py
catalyst/utils/visualization.py
catalyst/utils/criterion/__init__.py
catalyst/utils/meters/__init__.py
catalyst/utils/meters/apmeter.py
catalyst/utils/meters/aucmeter.py
catalyst/utils/meters/averagevaluemeter.py
catalyst/utils/meters/classerrormeter.py
catalyst/utils/meters/confusionmeter.py
catalyst/utils/meters/mapmeter.py
catalyst/utils/meters/meter.py
catalyst/utils/meters/movingaveragevaluemeter.py
catalyst/utils/meters/msemeter.py
catalyst/utils/meters/ppv_tpr_f1_meter.py
catalyst/utils/metrics/__init__.py
catalyst/utils/metrics/accuracy.py
catalyst/utils/metrics/dice.py
catalyst/utils/metrics/f1_score.py
catalyst/utils/metrics/focal.py
catalyst/utils/metrics/iou.py
catalyst/utils/tests/__init__.py
catalyst/utils/tests/registery_foo.py
catalyst/utils/tests/test_argparse.py
catalyst/utils/tests/test_config.py
catalyst/utils/tests/test_dataset.py
catalyst/utils/tests/test_hash.py
catalyst/utils/tests/test_image.py
catalyst/utils/tests/test_iou.py
catalyst/utils/tests/test_metric_manager.py
catalyst/utils/tests/test_misc.py
catalyst/utils/tests/test_pandas.py
catalyst/utils/tests/test_ppv_tpr_f1.py
catalyst/utils/tests/test_registry.py
catalyst/utils/tests/test_tensorboard.py
catalyst/utils/tools/__init__.py
catalyst/utils/tools/dynamic_array.py
catalyst/utils/tools/frozen_class.py
catalyst/utils/tools/metric_manager.py
catalyst/utils/tools/registry.py
catalyst/utils/tools/seeder.py
catalyst/utils/tools/tensorboard.py
catalyst/utils/tools/time_manager.py
catalyst/utils/tools/typing.py