OutHeadFC

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

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

forward

Inherited Methods

get_postprocessors
postprocess