clinicadl.losses.config.MultiMarginLossConfig

clinicadl.losses.config.MultiMarginLossConfig

Config class for torch.nn.MultiMarginLoss.

weight must be pass via a list and not via torch.Tensor.

parameter p: Order = 1

parameter margin: float = 1.0

parameter weight: Optional[List[NonNegativeFloat]] = None

parameter reduction: Reduction = 'mean'