clinicadl.networks.config.AutoEncoderConfig

clinicadl.networks.config.AutoEncoderConfig[source]

Config class for clinicadl.networks.nn.AutoEncoder.

parameter in_shape: Sequence[PositiveInt] [Required]
parameter latent_size: PositiveInt [Required]
parameter conv_args: ConvEncoderOptions [Required]
parameter mlp_args: MLPOptions = None
parameter out_channels: Optional[PositiveInt] = None
parameter output_act: Union[ActFunction, tuple[ActFunction, dict[str, Any]], None] = None
parameter unpooling_mode: UnpoolingMode = UnpoolingMode.NEAREST