OutHeadFC

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

Bases: OutHead

Fully-Connected Feed-Forward Network for decoding an unstrucured, 1D feature.

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

  • latent_dims (List[int]) – A list of integers representing the dimensions of the latent space. The length of the sequence defines the number of layers.

  • activation (Union[nn.Module, str]) – The activation function to use.

  • out_activation (Union[nn.Module, str], optional, default=None) – The activation function to use in the last layer.

Methods

Inherited Methods