clinicadl.networks.config.AttentionUNetConfig¶
- clinicadl.networks.config.AttentionUNetConfig[source]¶
Config class for
clinicadl.networks.nn.AttentionUNet.- parameter spatial_dims: PositiveInt [Required]¶
- parameter in_channels: PositiveInt [Required]¶
- parameter out_channels: PositiveInt [Required]¶
- parameter channels: Sequence[PositiveInt] = (64, 128, 256, 512, 1024)¶
- parameter act: ActivationParameters = ActFunction.RELU¶
- parameter output_act: Optional[ActivationParameters] = None¶
- parameter dropout: Optional[NonNegativeFloat] = None¶