clinicadl.losses.config.KLDivLossConfig

clinicadl.losses.config.KLDivLossConfig[source]

Config class for torch.nn.KLDivLoss.

parameter reduction: Reduction = 'mean'
parameter log_target: bool = False