Class

MPSCNNLossDescriptor

An object that specifies properties used by a loss kernel.

Declaration

class MPSCNNLossDescriptor : NSObject

Topics

Initializers

init(type: MPSCNNLossType, reductionType: MPSCNNReductionType)

Instance Properties

var delta: Float

var epsilon: Float

var labelSmoothing: Float

var lossType: MPSCNNLossType

var numberOfClasses: Int

var reductionType: MPSCNNReductionType

var weight: Float

Relationships

Inherits From

Conforms To

See Also

Loss Layers

class MPSCNNLoss

A kernel that computes the loss and loss gradient between specified predictions and labels.

class MPSCNNLossDataDescriptor

An object that specifies properties used by a loss data descriptor.

class MPSCNNLossLabels

A class that stores the per-element weight buffer used by loss and gradient loss kernels.

class MPSCNNYOLOLoss

A kernel that computes the YOLO loss and loss gradient between specified predictions and labels.

class MPSCNNYOLOLossDescriptor

An object that specifies properties used by a YOLO loss kernel.