Index

_

__init__() (catalyst.callbacks.batch_overfit.BatchOverfitCallback method) (catalyst.callbacks.batch_transform.BatchTransformCallback method) (catalyst.callbacks.checkpoint.CheckpointCallback method) (catalyst.callbacks.control_flow.ControlFlowCallback method) (catalyst.callbacks.criterion.CriterionCallback method) (catalyst.callbacks.metric.BatchMetricCallback method) (catalyst.callbacks.metric_aggregation.MetricAggregationCallback method) (catalyst.callbacks.metrics.accuracy.AccuracyCallback method) (catalyst.callbacks.metrics.accuracy.MultilabelAccuracyCallback method) (catalyst.callbacks.metrics.auc.AUCCallback method) (catalyst.callbacks.metrics.classification.MultilabelPrecisionRecallF1SupportCallback method) (catalyst.callbacks.metrics.classification.PrecisionRecallF1SupportCallback method) (catalyst.callbacks.metrics.cmc_score.CMCScoreCallback method) (catalyst.callbacks.metrics.cmc_score.ReidCMCScoreCallback method) (catalyst.callbacks.metrics.confusion_matrix.ConfusionMatrixCallback method) (catalyst.callbacks.metrics.functional_metric.FunctionalMetricCallback method) (catalyst.callbacks.metrics.recsys.HitrateCallback method) (catalyst.callbacks.metrics.recsys.MAPCallback method) (catalyst.callbacks.metrics.recsys.MRRCallback method) (catalyst.callbacks.metrics.recsys.NDCGCallback method) (catalyst.callbacks.metrics.segmentation.DiceCallback method) (catalyst.callbacks.metrics.segmentation.IOUCallback method) (catalyst.callbacks.metrics.segmentation.TrevskyCallback method) (catalyst.callbacks.misc.CheckRunCallback method) (catalyst.callbacks.misc.EarlyStoppingCallback method) (catalyst.callbacks.misc.TimerCallback method) (catalyst.callbacks.onnx.OnnxCallback method) (catalyst.callbacks.optimizer.OptimizerCallback method) (catalyst.callbacks.optuna.OptunaPruningCallback method) (catalyst.callbacks.periodic_loader.PeriodicLoaderCallback method) (catalyst.callbacks.pruning.PruningCallback method) (catalyst.callbacks.quantization.QuantizationCallback method) (catalyst.callbacks.scheduler.LRFinder method) (catalyst.callbacks.scheduler.SchedulerCallback method) (catalyst.callbacks.tracing.TracingCallback method) (catalyst.contrib.data.augmentor.Augmentor method) (catalyst.contrib.data.augmentor.AugmentorCompose method) (catalyst.contrib.data.augmentor.AugmentorKeys method) (catalyst.contrib.data.cv.dataset.ImageFolderDataset method) (catalyst.contrib.data.cv.reader.ImageReader method) (catalyst.contrib.data.cv.reader.MaskReader method) (catalyst.contrib.data.nifti.reader.NiftiReader method) (catalyst.contrib.data.reader.IReader method) (catalyst.contrib.data.reader.LambdaReader method) (catalyst.contrib.data.reader.ReaderCompose method) (catalyst.contrib.data.reader.ScalarReader method) (catalyst.contrib.datasets.cv.imagenette.Imagenette method) (catalyst.contrib.datasets.cv.imagenette.Imagenette160 method) (catalyst.contrib.datasets.cv.imagenette.Imagenette320 method) (catalyst.contrib.datasets.cv.imagewang.Imagewang method) (catalyst.contrib.datasets.cv.imagewang.Imagewang160 method) (catalyst.contrib.datasets.cv.imagewang.Imagewang320 method) (catalyst.contrib.datasets.cv.Imagewoof160 method) (catalyst.contrib.datasets.cv.Imagewoof320 method) (catalyst.contrib.datasets.mnist.MNIST method) (catalyst.contrib.datasets.movielens.MovieLens method) (catalyst.contrib.nn.criterion.circle.CircleLoss method) (catalyst.contrib.nn.criterion.dice.DiceLoss method) (catalyst.contrib.nn.criterion.focal.FocalLossBinary method) (catalyst.contrib.nn.criterion.focal.FocalLossMultiClass method) (catalyst.contrib.nn.criterion.iou.IoULoss method) (catalyst.contrib.nn.criterion.margin.MarginLoss method) (catalyst.contrib.nn.criterion.trevsky.FocalTrevskyLoss method) (catalyst.contrib.nn.criterion.trevsky.TrevskyLoss method) (catalyst.contrib.nn.criterion.triplet.TripletMarginLossWithSampler method) (catalyst.contrib.nn.criterion.wing.WingLoss method) (catalyst.contrib.nn.modules.se.cSE method) (catalyst.contrib.nn.modules.se.scSE method) (catalyst.contrib.nn.modules.se.sSE method) (catalyst.contrib.nn.optimizers.adamp.AdamP method) (catalyst.contrib.nn.optimizers.lamb.Lamb method) (catalyst.contrib.nn.optimizers.lookahead.Lookahead method) (catalyst.contrib.nn.optimizers.qhadamw.QHAdamW method) (catalyst.contrib.nn.optimizers.radam.RAdam method) (catalyst.contrib.nn.optimizers.ralamb.Ralamb method) (catalyst.contrib.nn.optimizers.sgdp.SGDP method) (catalyst.contrib.nn.schedulers.onecycle.OneCycleLRWithWarmup method) (catalyst.core.callback.Callback method) (catalyst.core.callback.CallbackList method) (catalyst.core.callback.CallbackWrapper method) (catalyst.data.AllTripletsSampler method) (catalyst.data.collate_fn.FilteringCollateFn method) (catalyst.data.dataset.metric_learning.MetricLearningTrainDataset method) (catalyst.data.dataset.metric_learning.QueryGalleryDataset method) (catalyst.data.dataset.torch.DatasetFromSampler method) (catalyst.data.dataset.torch.ListDataset method) (catalyst.data.dataset.torch.MergeDataset method) (catalyst.data.dataset.torch.NumpyDataset method) (catalyst.data.dataset.torch.PathsDataset method) (catalyst.data.HardTripletsSampler method) (catalyst.data.loader.BatchLimitLoaderWrapper method) (catalyst.data.loader.BatchPrefetchLoaderWrapper method) (catalyst.data.sampler.BalanceBatchSampler method) (catalyst.data.sampler.BalanceClassSampler method) (catalyst.data.sampler.DistributedSamplerWrapper method) (catalyst.data.sampler.DynamicBalanceClassSampler method) (catalyst.data.sampler.DynamicLenBatchSampler method) (catalyst.data.sampler.MiniEpochSampler method) (catalyst.data.transforms.Compose method) (catalyst.data.transforms.Normalize method) (catalyst.data.transforms.ToTensor method) (catalyst.metrics._additive.AdditiveValueMetric method) (catalyst.metrics._metric.AccumulationMetric method) (catalyst.metrics._metric.ICallbackBatchMetric method) (catalyst.metrics._metric.ICallbackLoaderMetric method) (catalyst.metrics._metric.IMetric method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.config.SupervisedConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.hydra.SupervisedHydraRunner method) (catalyst.runners.runner.Runner method) (catalyst.runners.runner.SupervisedRunner method) (catalyst.runners.supervised.ISupervisedRunner method) (catalyst.tools.forward_wrapper.ModelForwardWrapper method) (catalyst.tools.metric_handler.MetricHandler method) (catalyst.tools.registry.Registry method) (catalyst.tools.time_manager.TimeManager method)

A

AccumulationMetric (class in catalyst.metrics._metric) accuracy() (in module catalyst.metrics.functional._accuracy) AccuracyCallback (class in catalyst.callbacks.metrics.accuracy) AccuracyMetric (class in catalyst.metrics._accuracy) AdaCos (class in catalyst.contrib.nn.modules.cosface) AdamP (class in catalyst.contrib.nn.optimizers.adamp) add() (catalyst.tools.registry.Registry method) add_from_module() (catalyst.tools.registry.Registry method) AdditiveValueMetric (class in catalyst.metrics._additive) All (catalyst.core.callback.CallbackNode attribute) all (catalyst.core.callback.CallbackNode attribute) all() (catalyst.tools.registry.Registry method) all_gather() (in module catalyst.utils.distributed) AllTripletsSampler (class in catalyst.data)

AMPEngine (class in catalyst.engines.amp) any2device() (in module catalyst.utils.torch) APEXEngine (class in catalyst.engines.apex) ArcFace (class in catalyst.contrib.nn.modules.arcface) ArcMarginProduct (class in catalyst.contrib.nn.modules.arcmargin) args_are_not_none() (in module catalyst.utils.misc) auc() (in module catalyst.metrics.functional._auc) AUCCallback (class in catalyst.callbacks.metrics.auc) AUCMetric (class in catalyst.metrics._auc) Augmentor (class in catalyst.contrib.data.augmentor) AugmentorCompose (class in catalyst.contrib.data.augmentor) AugmentorKeys (class in catalyst.contrib.data.augmentor) autocast() (catalyst.core.engine.IEngine method) average_precision() (in module catalyst.metrics.functional._average_precision) average_weights() (in module catalyst.utils.swa)

B

backward_loss() (catalyst.core.engine.IEngine method) balance_classes() (in module catalyst.contrib.utils.pandas) BalanceBatchSampler (class in catalyst.data.sampler) BalanceClassSampler (class in catalyst.data.sampler) BatchLimitLoaderWrapper (class in catalyst.data.loader) BatchMetricCallback (class in catalyst.callbacks.metric)

BatchOverfitCallback (class in catalyst.callbacks.batch_overfit) BatchPrefetchLoaderWrapper (class in catalyst.data.loader) BatchTransformCallback (class in catalyst.callbacks.batch_transform) binary_auc() (in module catalyst.metrics.functional._auc) binary_average_precision() (in module catalyst.metrics.functional._average_precision) BinaryPrecisionRecallF1Metric (class in catalyst.metrics._classification) boolean_flag() (in module catalyst.utils.misc)

C

calc_lr() (catalyst.callbacks.scheduler.LRFinder method) calc_momentum() (catalyst.callbacks.scheduler.LRFinder method) Callback (class in catalyst.core.callback) CallbackList (class in catalyst.core.callback) CallbackNode (class in catalyst.core.callback) CallbackOrder (class in catalyst.core.callback) CallbackScope (class in catalyst.core.callback) CallbackWrapper (class in catalyst.core.callback) catalyst.callbacks module catalyst.callbacks.metrics module catalyst.contrib module catalyst.contrib.__main__ module catalyst.contrib.data module catalyst.contrib.datasets module catalyst.contrib.nn.modules.arcface module catalyst.contrib.nn.modules.arcmargin module catalyst.contrib.nn.modules.cosface module catalyst.contrib.nn.modules.curricularface module catalyst.contrib.nn.modules.softmax module catalyst.contrib.nn.schedulers module catalyst.contrib.utils module catalyst.contrib.utils.compression module catalyst.contrib.utils.image module catalyst.contrib.utils.pandas module catalyst.contrib.utils.parallel module catalyst.contrib.utils.serialization module catalyst.contrib.utils.thresholds module catalyst.contrib.utils.visualization module catalyst.core module catalyst.data module catalyst.data.dataset module catalyst.dl.__main__ module catalyst.metrics module catalyst.metrics.functional._accuracy module catalyst.metrics.functional._auc module catalyst.metrics.functional._average_precision module catalyst.metrics.functional._classification module catalyst.metrics.functional._cmc_score module catalyst.metrics.functional._f1_score module

catalyst.metrics.functional._focal module catalyst.metrics.functional._hitrate module catalyst.metrics.functional._misc module catalyst.metrics.functional._mrr module catalyst.metrics.functional._ndcg module catalyst.metrics.functional._precision module catalyst.metrics.functional._recall module catalyst.metrics.functional._segmentation module catalyst.utils module catalyst.utils.config module catalyst.utils.data module catalyst.utils.distributed module catalyst.utils.misc module catalyst.utils.numpy module catalyst.utils.onnx module catalyst.utils.pruning module catalyst.utils.quantization module catalyst.utils.swa module catalyst.utils.sys module catalyst.utils.torch module catalyst.utils.tracing module check_consistent_length() (in module catalyst.metrics.functional._misc) CheckpointCallback (class in catalyst.callbacks.checkpoint) CheckRunCallback (class in catalyst.callbacks.misc) CircleLoss (class in catalyst.contrib.nn.criterion.circle) close_log() (catalyst.core.logger.ILogger method) cmc_score() (in module catalyst.metrics.functional._cmc_score) cmc_score_count() (in module catalyst.metrics.functional._cmc_score) CMCMetric (class in catalyst.metrics._cmc_score) CMCScoreCallback (class in catalyst.callbacks.metrics.cmc_score) Compose (class in catalyst.data.transforms) compute() (catalyst.metrics._additive.AdditiveValueMetric method) (catalyst.metrics._metric.AccumulationMetric method) (catalyst.metrics._metric.IMetric method) compute_key_value() (catalyst.metrics._metric.AccumulationMetric method) (catalyst.metrics._metric.ICallbackBatchMetric method) (catalyst.metrics._metric.ICallbackLoaderMetric method) ConfigRunner (class in catalyst.runners.config) ConfusionMatrixCallback (class in catalyst.callbacks.metrics.confusion_matrix) ConfusionMatrixMetric (class in catalyst.metrics._confusion_matrix) ConsoleLogger (class in catalyst.loggers.console) ControlFlowCallback (class in catalyst.callbacks.control_flow) convert_labels2list() (in module catalyst.utils.misc) CosFace (class in catalyst.contrib.nn.modules.cosface) create_dataframe() (in module catalyst.contrib.utils.pandas) create_dataset() (in module catalyst.contrib.utils.pandas) CriterionCallback (class in catalyst.callbacks.criterion) cSE (class in catalyst.contrib.nn.modules.se) CSVLogger (class in catalyst.loggers.csv) CurricularFace (class in catalyst.contrib.nn.modules.curricularface)

D

dataframe_to_list() (in module catalyst.contrib.utils.pandas) DataParallelAMPEngine (class in catalyst.engines.amp) DataParallelApexEngine (class in catalyst.engines.apex) DataParallelEngine (class in catalyst.engines.torch) DatasetFromSampler (class in catalyst.data.dataset.torch) dcg() (in module catalyst.metrics.functional._ndcg) deinit_components() (catalyst.core.engine.IEngine method) deserialize() (in module catalyst.contrib.utils.serialization) detach_tensor() (in module catalyst.utils.torch) DeviceEngine (class in catalyst.engines.torch) dice() (in module catalyst.metrics.functional._segmentation)

DiceCallback (class in catalyst.callbacks.metrics.segmentation) DiceLoss (class in catalyst.contrib.nn.criterion.dice) DiceMetric (class in catalyst.metrics._segmentation) DistributedDataParallelAMPEngine (class in catalyst.engines.amp) DistributedDataParallelApexEngine (class in catalyst.engines.apex) DistributedDataParallelEngine (class in catalyst.engines.torch) DistributedSamplerWrapper (class in catalyst.data.sampler) dump_code() (in module catalyst.utils.sys) dump_environment() (in module catalyst.utils.sys) DynamicBalanceClassSampler (class in catalyst.data.sampler) DynamicLenBatchSampler (class in catalyst.data.sampler)

E

EarlyStoppingCallback (class in catalyst.callbacks.misc) Experiment (catalyst.core.callback.CallbackScope attribute) experiment (catalyst.core.callback.CallbackScope attribute)

External (catalyst.core.callback.CallbackOrder attribute) external (catalyst.core.callback.CallbackOrder attribute) external_extra (catalyst.core.callback.CallbackOrder attribute) ExternalExtra (catalyst.core.callback.CallbackOrder attribute)

F

f1_score() (in module catalyst.metrics.functional._f1_score) f1score() (in module catalyst.metrics.functional._classification) fbeta_score() (in module catalyst.metrics.functional._f1_score) FilteringCollateFn (class in catalyst.data.collate_fn) find_value_ids() (in module catalyst.utils.misc) flatten_dict() (in module catalyst.utils.misc) flush_log() (catalyst.core.logger.ILogger method) FocalLossBinary (class in catalyst.contrib.nn.criterion.focal) FocalLossMultiClass (class in catalyst.contrib.nn.criterion.focal) FocalTrevskyLoss (class in catalyst.contrib.nn.criterion.trevsky) folds_to_list() (in module catalyst.contrib.utils.pandas)

forward() (catalyst.contrib.nn.modules.arcface.ArcFace method) (catalyst.contrib.nn.modules.arcface.SubCenterArcFace method) (catalyst.contrib.nn.modules.arcmargin.ArcMarginProduct method) (catalyst.contrib.nn.modules.cosface.AdaCos method) (catalyst.contrib.nn.modules.cosface.CosFace method) (catalyst.contrib.nn.modules.curricularface.CurricularFace method) (catalyst.contrib.nn.modules.softmax.SoftMax method) (catalyst.runners.supervised.ISupervisedRunner method) (catalyst.tools.forward_wrapper.ModelForwardWrapper method) FrozenClass (class in catalyst.tools.frozen_class) FunctionalBatchMetric (class in catalyst.metrics._functional_metric) FunctionalMetricCallback (class in catalyst.callbacks.metrics.functional_metric)

G

gallery_size() (catalyst.data.dataset.metric_learning.QueryGalleryDataset property) get() (catalyst.tools.registry.Registry method) get_aggregated_metrics() (in module catalyst.metrics.functional._classification) get_attr() (in module catalyst.utils.misc) get_available_engine() (in module catalyst.utils.torch) get_available_gpus() (in module catalyst.utils.torch) get_averaged_weights_by_path_mask() (in module catalyst.utils.swa) get_baseline_thresholds() (in module catalyst.contrib.utils.thresholds) get_best_multiclass_thresholds() (in module catalyst.contrib.utils.thresholds) get_best_multilabel_thresholds() (in module catalyst.contrib.utils.thresholds) get_binary_metrics() (in module catalyst.metrics.functional._classification) get_binary_statistics() (in module catalyst.metrics.functional._misc) get_binary_threshold() (in module catalyst.contrib.utils.thresholds) get_binary_threshold_cv() (in module catalyst.contrib.utils.thresholds) get_by_keys() (in module catalyst.utils.misc) get_callbacks() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) (catalyst.runners.runner.SupervisedRunner method) get_config_runner() (in module catalyst.utils.sys) get_criterion() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_dataset_labeling() (in module catalyst.contrib.utils.pandas) get_datasets() (catalyst.core.runner.IRunner method) get_default_topk_args() (in module catalyst.metrics.functional._misc) get_device() (in module catalyst.utils.torch) get_dictkey_auto_fn() (in module catalyst.utils.misc) get_distributed_params() (in module catalyst.utils.distributed) get_engine() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_fn_argsnames() (in module catalyst.utils.misc) get_fn_default_params() (in module catalyst.utils.misc) get_from_params() (catalyst.tools.registry.Registry method) get_hash() (in module catalyst.utils.misc) get_if_str() (catalyst.tools.registry.Registry method) get_instance() (catalyst.tools.registry.Registry method) get_labels() (catalyst.data.dataset.metric_learning.MetricLearningTrainDataset method) get_loader() (in module catalyst.utils.data) get_loaders() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method)

get_loaders_from_params() (in module catalyst.utils.data) get_loggers() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_model() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_multiclass_statistics() (in module catalyst.metrics.functional._misc) get_multiclass_thresholds() (in module catalyst.contrib.utils.thresholds) get_multiclass_thresholds_greedy() (in module catalyst.contrib.utils.thresholds) get_multilabel_statistics() (in module catalyst.metrics.functional._misc) get_multilabel_thresholds() (in module catalyst.contrib.utils.thresholds) get_multilabel_thresholds_cv() (in module catalyst.contrib.utils.thresholds) get_multilabel_thresholds_greedy() (in module catalyst.contrib.utils.thresholds) get_network_output() (in module catalyst.utils.torch) get_nn_from_ddp_module() (in module catalyst.utils.distributed) get_one_hot() (in module catalyst.utils.numpy) get_optimal_inner_init() (in module catalyst.utils.torch) get_optimizable_params() (in module catalyst.utils.torch) get_optimizer() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_optimizer_momentum() (in module catalyst.utils.torch) get_optimizer_momentum_list() (in module catalyst.utils.torch) get_pool() (in module catalyst.contrib.utils.parallel) get_pruning_fn() (in module catalyst.utils.pruning) get_rank() (in module catalyst.utils.distributed) get_requires_grad() (in module catalyst.utils.torch) get_scheduler() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_segmentation_statistics() (in module catalyst.metrics.functional._segmentation) get_short_hash() (in module catalyst.utils.misc) get_stage_len() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_thresholds_greedy() (in module catalyst.contrib.utils.thresholds) get_trial() (catalyst.core.runner.IRunner method) (catalyst.runners.config.ConfigRunner method) (catalyst.runners.hydra.HydraRunner method) (catalyst.runners.runner.Runner method) get_utcnow_time() (in module catalyst.utils.misc)

H

handle_batch() (catalyst.core.runner.IRunner method) (catalyst.runners.supervised.ISupervisedRunner method) handle_score_is_better() (catalyst.callbacks.misc.EarlyStoppingCallback method) handle_score_is_not_better() (catalyst.callbacks.misc.EarlyStoppingCallback method) HardClusterSampler (class in catalyst.data) HardTripletsSampler (class in catalyst.data) has_image_extension() (in module catalyst.contrib.utils.image)

hitrate() (in module catalyst.metrics.functional._hitrate) HitrateCallback (class in catalyst.callbacks.metrics.recsys) HitrateMetric (class in catalyst.metrics._hitrate) hparams() (catalyst.core.runner.IRunner property) (catalyst.runners.config.ConfigRunner property) (catalyst.runners.hydra.HydraRunner property) (catalyst.runners.runner.Runner property) HydraRunner (class in catalyst.runners.hydra)

I

ICallback (class in catalyst.core.callback) ICallbackBatchMetric (class in catalyst.metrics._metric) ICallbackLoaderMetric (class in catalyst.metrics._metric) IEngine (class in catalyst.core.engine) IInbatchTripletSampler (class in catalyst.data) ILogger (class in catalyst.core.logger) ImageFolderDataset (class in catalyst.contrib.data.cv.dataset) Imagenette (class in catalyst.contrib.datasets.cv.imagenette) Imagenette160 (class in catalyst.contrib.datasets.cv.imagenette) Imagenette320 (class in catalyst.contrib.datasets.cv.imagenette) ImageReader (class in catalyst.contrib.data.cv.reader) Imagewang (class in catalyst.contrib.datasets.cv.imagewang) Imagewang160 (class in catalyst.contrib.datasets.cv.imagewang) Imagewang320 (class in catalyst.contrib.datasets.cv.imagewang) imagewoof (in module catalyst.contrib.datasets.cv) Imagewoof160 (class in catalyst.contrib.datasets.cv) Imagewoof320 (class in catalyst.contrib.datasets.cv) IMetric (class in catalyst.metrics._metric) import_module() (in module catalyst.utils.sys)

imread() (in module catalyst.contrib.utils.image) imsave() (in module catalyst.contrib.utils.image) imwrite() (in module catalyst.contrib.utils.image) InBatchTripletsSampler (class in catalyst.data) init_components() (catalyst.core.engine.IEngine method) Internal (catalyst.core.callback.CallbackOrder attribute) internal (catalyst.core.callback.CallbackOrder attribute) iou() (in module catalyst.metrics.functional._segmentation) IOUCallback (class in catalyst.callbacks.metrics.segmentation) IoULoss (class in catalyst.contrib.nn.criterion.iou) IOUMetric (class in catalyst.metrics._segmentation) IReader (class in catalyst.contrib.data.reader) IRunner (class in catalyst.core.runner) is_ddp() (catalyst.core.engine.IEngine property) is_exception() (in module catalyst.utils.misc) is_master_process() (catalyst.core.engine.IEngine property) is_worker_process() (catalyst.core.engine.IEngine property) ISupervisedRunner (class in catalyst.runners.supervised) ITrial (class in catalyst.core.trial)

L

Lamb (class in catalyst.contrib.nn.optimizers.lamb) LambdaReader (class in catalyst.contrib.data.reader) late_add() (catalyst.tools.registry.Registry method) len() (catalyst.tools.registry.Registry method) ListDataset (class in catalyst.data.dataset.torch) load_checkpoint() (catalyst.core.engine.IEngine method) (in module catalyst.utils.torch) load_config() (in module catalyst.utils.config) LoaderMetricCallback (class in catalyst.callbacks.metric) log_artifact() (catalyst.core.logger.ILogger method)

log_hparams() (catalyst.core.logger.ILogger method) (catalyst.core.runner.IRunner method) log_image() (catalyst.core.logger.ILogger method) (catalyst.core.runner.IRunner method) log_metrics() (catalyst.core.logger.ILogger method) (catalyst.core.runner.IRunner method) logdir() (catalyst.runners.config.ConfigRunner property) (catalyst.runners.hydra.HydraRunner property) Lookahead (class in catalyst.contrib.nn.optimizers.lookahead) LRFinder (class in catalyst.callbacks.scheduler)

M

make_batch_step() (catalyst.callbacks.scheduler.SchedulerCallback method) make_epoch_step() (catalyst.callbacks.scheduler.SchedulerCallback method) make_tuple() (in module catalyst.utils.misc) map_dataframe() (in module catalyst.contrib.utils.pandas) MAPCallback (class in catalyst.callbacks.metrics.recsys) MAPMetric (class in catalyst.metrics._map) MarginLoss (class in catalyst.contrib.nn.criterion.margin) masked_cmc_score() (in module catalyst.metrics.functional._cmc_score) MaskReader (class in catalyst.contrib.data.cv.reader) Master (catalyst.core.callback.CallbackNode attribute) master (catalyst.core.callback.CallbackNode attribute) maybe_recursive_call() (in module catalyst.utils.misc) mean_average_precision() (in module catalyst.metrics.functional._average_precision) mean_reduce() (in module catalyst.utils.distributed) merge_dicts() (in module catalyst.utils.misc) MergeDataset (class in catalyst.data.dataset.torch) Metric (catalyst.core.callback.CallbackOrder attribute) metric (catalyst.core.callback.CallbackOrder attribute) metric_aggregation (catalyst.core.callback.CallbackOrder attribute) MetricAggregation (catalyst.core.callback.CallbackOrder attribute) MetricAggregationCallback (class in catalyst.callbacks.metric_aggregation) MetricHandler (class in catalyst.tools.metric_handler) MetricLearningTrainDataset (class in catalyst.data.dataset.metric_learning) mimread() (in module catalyst.contrib.utils.image) MiniEpochSampler (class in catalyst.data.sampler) MLflowLogger (class in catalyst.loggers.mlflow) MNIST (class in catalyst.contrib.datasets.mnist) ModelForwardWrapper (class in catalyst.tools.forward_wrapper) module catalyst.callbacks catalyst.callbacks.metrics catalyst.contrib catalyst.contrib.__main__ catalyst.contrib.data catalyst.contrib.datasets catalyst.contrib.nn.modules.arcface catalyst.contrib.nn.modules.arcmargin catalyst.contrib.nn.modules.cosface catalyst.contrib.nn.modules.curricularface catalyst.contrib.nn.modules.softmax catalyst.contrib.nn.schedulers catalyst.contrib.utils catalyst.contrib.utils.compression catalyst.contrib.utils.image catalyst.contrib.utils.pandas catalyst.contrib.utils.parallel catalyst.contrib.utils.serialization catalyst.contrib.utils.thresholds catalyst.contrib.utils.visualization catalyst.core catalyst.data catalyst.data.dataset catalyst.dl.__main__ catalyst.metrics catalyst.metrics.functional._accuracy catalyst.metrics.functional._auc catalyst.metrics.functional._average_precision catalyst.metrics.functional._classification catalyst.metrics.functional._cmc_score catalyst.metrics.functional._f1_score catalyst.metrics.functional._focal catalyst.metrics.functional._hitrate catalyst.metrics.functional._misc catalyst.metrics.functional._mrr catalyst.metrics.functional._ndcg catalyst.metrics.functional._precision catalyst.metrics.functional._recall catalyst.metrics.functional._segmentation catalyst.utils catalyst.utils.config catalyst.utils.data catalyst.utils.distributed catalyst.utils.misc catalyst.utils.numpy catalyst.utils.onnx catalyst.utils.pruning catalyst.utils.quantization catalyst.utils.swa catalyst.utils.sys catalyst.utils.torch catalyst.utils.tracing

MovieLens (class in catalyst.contrib.datasets.movielens) mrr() (in module catalyst.metrics.functional._mrr) MRRCallback (class in catalyst.callbacks.metrics.recsys) MRRMetric (class in catalyst.metrics._mrr) MulticlassPrecisionRecallF1SupportMetric (class in catalyst.metrics._classification) multilabel_accuracy() (in module catalyst.metrics.functional._accuracy) MultilabelAccuracyCallback (class in catalyst.callbacks.metrics.accuracy) MultilabelAccuracyMetric (class in catalyst.metrics._accuracy) MultilabelPrecisionRecallF1SupportCallback (class in catalyst.callbacks.metrics.classification) MultilabelPrecisionRecallF1SupportMetric (class in catalyst.metrics._classification)

N

name() (catalyst.runners.config.ConfigRunner property) (catalyst.runners.hydra.HydraRunner property) (catalyst.runners.runner.Runner property) ndcg() (in module catalyst.metrics.functional._ndcg) NDCGCallback (class in catalyst.callbacks.metrics.recsys)

NDCGMetric (class in catalyst.metrics._ndcg) NeptuneLogger (class in catalyst.loggers.neptune) NiftiReader (class in catalyst.contrib.data.nifti.reader) Normalize (class in catalyst.data.transforms) NumpyDataset (class in catalyst.data.dataset.torch)

O

on_batch_end() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_batch_start() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_epoch_end() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_epoch_start() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_exception() (catalyst.callbacks.misc.TqdmCallback method) (catalyst.core.callback.CallbackList method) (catalyst.core.callback.CallbackWrapper method) (catalyst.core.callback.ICallback method) on_experiment_end() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_experiment_start() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method)

on_loader_end() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_loader_start() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_stage_end() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) on_stage_start() (catalyst.core.callback.CallbackList method) (catalyst.core.callback.ICallback method) OneCycleLRWithWarmup (class in catalyst.contrib.nn.schedulers.onecycle) onnx_export() (in module catalyst.utils.onnx) OnnxCallback (class in catalyst.callbacks.onnx) Optimizer (catalyst.core.callback.CallbackOrder attribute) optimizer (catalyst.core.callback.CallbackOrder attribute) optimizer_step() (catalyst.core.engine.IEngine method) OptimizerCallback (class in catalyst.callbacks.optimizer) OptunaPruningCallback (class in catalyst.callbacks.optuna) outer_init() (in module catalyst.utils.torch)

P

pack() (in module catalyst.contrib.utils.compression) pack_checkpoint() (catalyst.core.engine.IEngine method) (in module catalyst.utils.torch) pack_if_needed() (in module catalyst.contrib.utils.compression) pairwise() (in module catalyst.utils.misc) parallel_imap() (in module catalyst.contrib.utils.parallel) PathsDataset (class in catalyst.data.dataset.torch) PeriodicLoaderCallback (class in catalyst.callbacks.periodic_loader) plot_confusion_matrix() (in module catalyst.contrib.utils.visualization) precision() (in module catalyst.metrics.functional._classification) (in module catalyst.metrics.functional._precision)

precision_recall_fbeta_support() (in module catalyst.metrics.functional._classification) PrecisionRecallF1SupportCallback (class in catalyst.callbacks.metrics.classification) predict_batch() (catalyst.runners.runner.Runner method) (catalyst.runners.runner.SupervisedRunner method) predict_loader() (catalyst.runners.runner.Runner method) prepare_cudnn() (in module catalyst.utils.torch) process_model_params() (in module catalyst.utils.torch) process_multilabel_components() (in module catalyst.metrics.functional._misc) process_recsys_components() (in module catalyst.metrics.functional._misc) prune_model() (in module catalyst.utils.pruning) PruningCallback (class in catalyst.callbacks.pruning)

Q

QHAdamW (class in catalyst.contrib.nn.optimizers.qhadamw) QuantizationCallback (class in catalyst.callbacks.quantization) quantize_model() (in module catalyst.utils.quantization)

quantize_onnx_model() (in module catalyst.utils.onnx) query_size() (catalyst.data.dataset.metric_learning.QueryGalleryDataset property) QueryGalleryDataset (class in catalyst.data.dataset.metric_learning)

R

RAdam (class in catalyst.contrib.nn.optimizers.radam) Ralamb (class in catalyst.contrib.nn.optimizers.ralamb) rank() (catalyst.core.engine.IEngine property) ReaderCompose (class in catalyst.contrib.data.reader) recall() (in module catalyst.metrics.functional._classification) (in module catalyst.metrics.functional._recall) reciprocal_rank() (in module catalyst.metrics.functional._mrr) reduced_focal_loss() (in module catalyst.metrics.functional._focal) RegionBasedMetric (class in catalyst.metrics._segmentation) Registry (class in catalyst.tools.registry) ReidCMCMetric (class in catalyst.metrics._cmc_score)

ReidCMCScoreCallback (class in catalyst.callbacks.metrics.cmc_score) remove_reparametrization() (in module catalyst.utils.pruning) render_figure_to_array() (in module catalyst.contrib.utils.visualization) reset() (catalyst.metrics._additive.AdditiveValueMetric method) (catalyst.metrics._metric.AccumulationMetric method) (catalyst.metrics._metric.ICallbackLoaderMetric method) (catalyst.metrics._metric.IMetric method) (catalyst.tools.time_manager.TimeManager method) reset_weights_if_possible() (in module catalyst.utils.torch) run() (catalyst.core.runner.IRunner method) Runner (class in catalyst.runners.runner) RunnerException (class in catalyst.core.runner)

S

sample() (catalyst.data.HardClusterSampler method) (catalyst.data.IInbatchTripletSampler method) (catalyst.data.InBatchTripletsSampler method) save_checkpoint() (catalyst.core.engine.IEngine method) (in module catalyst.utils.torch) save_config() (in module catalyst.utils.config) ScalarReader (class in catalyst.contrib.data.reader) Scheduler (catalyst.core.callback.CallbackOrder attribute) scheduler (catalyst.core.callback.CallbackOrder attribute) SchedulerCallback (class in catalyst.callbacks.scheduler) scSE (class in catalyst.contrib.nn.modules.se) seed() (catalyst.core.runner.IRunner property) (catalyst.runners.config.ConfigRunner property) (catalyst.runners.hydra.HydraRunner property) (catalyst.runners.runner.Runner property) separate_tags() (in module catalyst.contrib.utils.pandas) serialize() (in module catalyst.contrib.utils.serialization) set_global_seed() (in module catalyst.utils.misc) set_optimizer_momentum() (in module catalyst.utils.torch) set_requires_grad() (in module catalyst.utils.torch) SGDP (class in catalyst.contrib.nn.optimizers.sgdp) sigmoid_focal_loss() (in module catalyst.metrics.functional._focal)

SoftMax (class in catalyst.contrib.nn.modules.softmax) split_dataframe() (in module catalyst.contrib.utils.pandas) split_dataframe_on_column_folds() (in module catalyst.contrib.utils.pandas) split_dataframe_on_folds() (in module catalyst.contrib.utils.pandas) split_dataframe_on_stratified_folds() (in module catalyst.contrib.utils.pandas) split_dataframe_train_test() (in module catalyst.contrib.utils.pandas) split_dataset_train_test() (in module catalyst.contrib.utils.pandas) sSE (class in catalyst.contrib.nn.modules.se) Stage (catalyst.core.callback.CallbackScope attribute) stage (catalyst.core.callback.CallbackScope attribute) stages() (catalyst.core.runner.IRunner property) (catalyst.runners.config.ConfigRunner property) (catalyst.runners.hydra.HydraRunner property) (catalyst.runners.runner.Runner property) start() (catalyst.tools.time_manager.TimeManager method) stop() (catalyst.tools.time_manager.TimeManager method) SubCenterArcFace (class in catalyst.contrib.nn.modules.arcface) sum_reduce() (in module catalyst.utils.distributed) SupervisedConfigRunner (class in catalyst.runners.config) SupervisedHydraRunner (class in catalyst.runners.hydra) SupervisedRunner (class in catalyst.runners.runner) sync_device() (catalyst.core.engine.IEngine method) sync_tensor() (catalyst.core.engine.IEngine method)

T

TensorboardLogger (class in catalyst.loggers.tensorboard) TimeManager (class in catalyst.tools.time_manager) TimerCallback (class in catalyst.callbacks.misc) ToTensor (class in catalyst.data.transforms) tqdm_parallel_imap() (in module catalyst.contrib.utils.parallel) TqdmCallback (class in catalyst.callbacks.misc) trace_model() (in module catalyst.utils.tracing) TracingCallback (class in catalyst.callbacks.tracing) train() (catalyst.runners.runner.Runner method) training (catalyst.contrib.nn.modules.arcface.ArcFace attribute) (catalyst.contrib.nn.modules.arcface.SubCenterArcFace attribute) (catalyst.contrib.nn.modules.arcmargin.ArcMarginProduct attribute) (catalyst.contrib.nn.modules.cosface.AdaCos attribute) (catalyst.contrib.nn.modules.cosface.CosFace attribute) (catalyst.contrib.nn.modules.curricularface.CurricularFace attribute) (catalyst.contrib.nn.modules.softmax.SoftMax attribute) (catalyst.tools.forward_wrapper.ModelForwardWrapper attribute)

trevsky() (in module catalyst.metrics.functional._segmentation) TrevskyCallback (class in catalyst.callbacks.metrics.segmentation) TrevskyLoss (class in catalyst.contrib.nn.criterion.trevsky) TrevskyMetric (class in catalyst.metrics._segmentation) trim_tensors() (in module catalyst.utils.torch) TripletMarginLossWithSampler (class in catalyst.contrib.nn.criterion.triplet)

U

unpack() (in module catalyst.contrib.utils.compression) unpack_checkpoint() (catalyst.core.engine.IEngine method) (in module catalyst.utils.torch) unpack_if_needed() (in module catalyst.contrib.utils.compression)

update() (catalyst.metrics._additive.AdditiveValueMetric method) (catalyst.metrics._metric.AccumulationMetric method) (catalyst.metrics._metric.ICallbackLoaderMetric method) (catalyst.metrics._metric.IMetric method) update_key_value() (catalyst.metrics._metric.ICallbackBatchMetric method)

W

WandbLogger (class in catalyst.loggers.wandb) WingLoss (class in catalyst.contrib.nn.criterion.wing)

Worker (catalyst.core.callback.CallbackNode attribute) worker (catalyst.core.callback.CallbackNode attribute) world_size() (catalyst.core.engine.IEngine property)

Z

zero_grad() (catalyst.core.engine.IEngine method)