braindecode.modules.GatherExcite#

class braindecode.modules.GatherExcite(in_channels: int, seq_len: int = 62, extra_params: bool = False, use_mlp: bool = False, reduction_rate: int = 4)[source]#

Gather-Excite Networks from [Hu2018b].

Parameters:
  • in_channels (int) – number of input feature channels

  • seq_len (int, default=62) – sequence length along temporal dimension

  • extra_params (bool, default=False) – whether to use a convolutional layer as a gather module

  • use_mlp (bool, default=False) – whether to use an excite block with fully-connected layers

  • reduction_rate (int, default=4) – reduction ratio of the excite block (if used)

References

[Hu2018b]

Hu, J., Albanie, S., Sun, G., Vedaldi, A., 2018.

Gather-Excite: Exploiting Feature Context in Convolutional Neural Networks. NeurIPS 2018.

Methods

forward(x)[source]#

Apply the Gather-Excite Networks block to the input tensor.

Parameters:

x (Pytorch.Tensor)

Return type:

Pytorch.Tensor