composer.algorithms.squeeze_excite

Modules

composer.algorithms.squeeze_excite.squeeze_excite

composer.algorithms.squeeze_excite.squeeze_excite

Adds Squeeze-and-Excitation blocks (Hu et al, 2019) after the Conv2d modules in a neural network.

See SqueezeExcite or the Method Card for details.

Functions

apply_squeeze_excite

Adds Squeeze-and-Excitation blocks (Hu et al, 2019) after torch.nn.Conv2d layers.

Classes

SqueezeExcite	Adds Squeeze-and-Excitation blocks (Hu et al, 2019) after the torch.nn.Conv2d modules in a neural network.
SqueezeExcite2d	Squeeze-and-Excitation block from (Hu et al, 2019)
SqueezeExciteConv2d	Helper class used to add a SqueezeExcite2d module after a torch.nn.Conv2d.