옵티마이저

public protocol Optimizer : CopyableToDevice

수치 최적화 프로그램.

최적화 프로그램은 최적화 알고리즘을 적용하여 미분 가능한 모델을 업데이트합니다.

  • 최적화할 모델의 유형입니다.

    선언

    associatedtype Model : Differentiable

  • 스칼라 매개변수 유형입니다.

    선언

    associatedtype Scalar : FloatingPoint

  • 학습률입니다.

    선언

    var learningRate: Scalar { get set }

  • 주어진 방향을 따라 주어진 모델을 업데이트합니다.

    선언

    mutating func update(_ model: inout Model, along direction: Model.TangentVector)