Contents

init(learningRate:gradientRescale:appliesGradientClipping:gradientClipMax:gradientClipMin:regularizationType:regularizationScale:)

Creates a descriptor with the learning rate, gradient rescale, clipping option and values, and regularization type and scale that you specify.

Declaration

convenience init(learningRate: Float, gradientRescale: Float, appliesGradientClipping: Bool, gradientClipMax: Float, gradientClipMin: Float, regularizationType: MLCRegularizationType, regularizationScale: Float)

Parameters

  • learningRate:

    The learning rate.

  • gradientRescale:

    The gradient rescale value.

  • appliesGradientClipping:

    A Boolean value that indicates whether you apply gradient clipping.

  • gradientClipMax:

    The maximum gradient value before the optimizer rescales the gradient, if you enable gradient clipping.

  • gradientClipMin:

    The minimum gradient value before the optimizer rescales the gradient, if you enable gradient clipping.

  • regularizationType:

    The regularization type.

  • regularizationScale:

    The regularization scale.

See Also

Creating an Optimizer Descriptor