open3d.ml.torch.modules.losses.SmoothL1Loss#

class open3d.ml.torch.modules.losses.SmoothL1Loss(beta=1.0, loss_weight=1.0)#

Smooth L1 loss.

Parameters:

beta (float, optional) – The threshold in the piecewise function. Defaults to 1.0.
reduction (str, optional) – The method to reduce the loss. Options are “none”, “mean” and “sum”. Defaults to “mean”.
loss_weight (float, optional) – The weight of loss.

__init__(beta=1.0, loss_weight=1.0)#: Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(pred, target, weight=None, avg_factor=None, **kwargs)#

Forward function.

Parameters:

pred (torch.Tensor) – The prediction.
target (torch.Tensor) – The learning target of the prediction.
weight (torch.Tensor, optional) – The weight of loss for each prediction. Defaults to None.
avg_factor (int, optional) – Average factor that is used to average the loss. Defaults to None.
reduction_override (str, optional) – The reduction method used to override the original reduction method of the loss. Defaults to None.