composer.algorithms.stochastic_depth

Modules

composer.algorithms.stochastic_depth.sample_stochastic_layers	composer.algorithms.stochastic_depth.sample_stochastic_layers
composer.algorithms.stochastic_depth.stochastic_depth	composer.algorithms.stochastic_depth.stochastic_depth
composer.algorithms.stochastic_depth.stochastic_layers	composer.algorithms.stochastic_depth.stochastic_layers

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

SampleStochasticBottleneck	Sample-wise stochastic ResNet Bottleneck block.
StochasticBottleneck	Stochastic ResNet Bottleneck block.
StochasticDepth	Applies Stochastic Depth (Huang et al, 2016) to the specified model.