SEBlock

class torchelie.nn.SEBlock(in_ch: int, reduction: int = 16)

A Squeeze-And-Excite block

Parameters

• in_ch (int) – input channels

• reduction (int) – channels reduction factor for the hidden number of channels

forward(x: torch.Tensor) → torch.Tensor

training: bool