clinicadl.networks.config.ConvEncoderConfig

clinicadl.networks.config.ConvEncoderConfig

Config class for clinicadl.networks.nn.ConvEncoder.

parameter spatial_dims: PositiveInt [Required]

parameter in_channels: PositiveInt [Required]

parameter channels: Sequence[PositiveInt] [Required]

parameter kernel_size: ConvParameters = 3

parameter stride: ConvParameters = 1

parameter padding: ConvParameters = 0

parameter dilation: ConvParameters = 1

parameter pooling: Optional[PoolingParameters] = (PoolingLayer.MAX, {'kernel_size': 2})

parameter pooling_indices: Sequence[int] = None

parameter act: Optional[ActivationParameters] = ActFunction.PRELU

parameter output_act: Optional[ActivationParameters] = None

parameter norm: Optional[ConvNormalizationParameters] = ConvNormLayer.INSTANCE

parameter dropout: Optional[NonNegativeFloat] = None

parameter bias: bool = True

parameter adn_ordering: str = 'NDA'