ResBlockFC

class aixd.mlmodel.architecture.blocks.ResBlockFC(*args: Any, **kwargs: Any)[source]

Bases: Module

Module implementing a fully-connected residual block.

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

  • out_channels (int) – The number of output channels.

  • activation (Union[nn.Module, str], optional, default=’leaky_relu’) – The activation function to use.

  • dropout (bool, optional, default=True) – Whether to use dropout or not.

  • batchnorm (bool, optional, default=True) – Whether to use batch normalization or not.

  • residual (bool, optional, default=True) – Whether to use a residual connection or not.

Methods