clinicadl.networks.config.VAEConfig¶

clinicadl.networks.config.VAEConfig[source]¶

Config class for clinicadl.networks.nn.VAE.

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: Optional[ActivationParameters] = None¶

parameter unpooling_mode: UnpoolingMode = UnpoolingMode.NEAREST¶