combustion.nn

Extensions to torch.nn, ranging from fundamental layers up to larger building blocks.

Activation Functions

Convolution Layers

Dropout Layers

Larger Modules

BiFPN

class combustion.nn.BiFPN

Alias for combustion.nn.BiFPN2d

Warning

This class is deprecated. Please use combustion.nn.BiFPN2d instead

class combustion.nn.BiFPN1d

1d variant of combustion.nn.BiFPN2d

class combustion.nn.BiFPN3d

3d variant of combustion.nn.BiFPN2d

Global Attention Upsample

class combustion.nn.AttentionUpsample1d

1d version of combustion.nn.AttentionUpsample2d.

class combustion.nn.AttentionUpsample3d

3d version of combustion.nn.AttentionUpsample2d.

MobileNetV3 Inverted Bottleneck

class combustion.nn.MobileNetConvBlock1d

1d version of combustion.nn.MobileNetConvBlock2d.

class combustion.nn.MobileNetConvBlock3d

3d version of combustion.nn.MobileNetConvBlock2d.

Object Contextual Representation

Squeeze and Excitation

Reduced Atrous Spatial Pyramid Pooling (R-ASPP Lite)

class combustion.nn.RASPPLite1d

1d version of combustion.nn.RASPPLite2d.

class combustion.nn.RASPPLite3d

3d version of combustion.nn.RASPPLite2d.

Loss Functions

Focal Loss

Utilities