composer.algorithms.stochastic_depth

Modules

composer.algorithms.stochastic_depth.stochastic_depth	Modules and layers for applying the Stochastic Depth algorithm.
composer.algorithms.stochastic_depth.stochastic_layers	Stochastic forward functions for ResNet Bottleneck modules.

Implements stochastic depth (Huang et al, 2016) for ResNet blocks.

See StochasticDepth, the sample-wise stochastic depth method card, or the block-wise stochastic depth method card for details.

Functions

apply_stochastic_depth

Applies Stochastic Depth (Huang et al, 2016) to the specified model.

Classes

StochasticDepth

Applies Stochastic Depth (Huang et al, 2016) to the specified model.