InHeadFC

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

Bases: InHead

Fully-Connected Feed-Forward Network for encoding 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.

Methods