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