composer.algorithms.algorithm_registry#

composer.algorithms.algorithm_registry

Functions

get_algorithm

composer.algorithms.algorithm_registry.get_algorithm

get_algorithm_registry

composer.algorithms.algorithm_registry.get_algorithm_registry

list_algorithms

composer.algorithms.algorithm_registry.list_algorithms

Classes

Algorithm

Base class for algorithms.

Hparams

These classes are used with yahp for YAML-based configuration.

AGCHparams

See AGC

AlgorithmHparams

Hyperparameters for algorithms.

AlibiHparams

See Alibi

AugMixHparams

See AugMix

BlurPoolHparams

See BlurPool

ChannelsLastHparams

ChannelsLast has no hyperparameters, so this class has no member variables.

ColOutHparams

See ColOut

CutMixHparams

See CutMix

CutOutHparams

See CutOut

EMAHparams

See EMA

FactorizeHparams

See Factorize

GhostBatchNormHparams

See GhostBatchNorm

LabelSmoothingHparams

See LabelSmoothing

LayerFreezingHparams

See LayerFreezing

MixUpHparams

See MixUp

NoOpModelHparams

composer.algorithms.hparams.NoOpModelHparams

ProgressiveResizingHparams

See ProgressiveResizing

RandAugmentHparams

See RandAugment

SAMHparams

See SAM

SWAHparams

See SWA

ScaleScheduleHparams

See ScaleSchedule

SelectiveBackpropHparams

See SelectiveBackprop

SeqLengthWarmupHparams

composer.algorithms.hparams.SeqLengthWarmupHparams

SqueezeExciteHparams

See SqueezeExcite

StochasticDepthHparams

See StochasticDepth

Attributes

  • Dict

  • List

  • Type

  • registry