OutHeadConv

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

Bases: OutHead

Convolutional Network for decoding spatially strucured data in 1, 2 or 3 dimensions.

Parameters:

• dim (int) – The dimensionality of the data: 1, 2 or 3.

• in_shape (Tuple[int]) – The shape of the input data.

• target_shape (Tuple[int]) – Final shape that the output data should have.

• 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]) – The activation function to use in the last layer.

• attn_block_indices (List[int]) – Sequence of indices where to insert an attention layer in the encoding blocks.

Methods

forward

Inherited Methods

get_postprocessors
postprocess