InHeadFC

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

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

forward