RAdam

public class RAdam<Model: Differentiable>: Optimizer
where
  Model.TangentVector: VectorProtocol & PointwiseMultiplicative & ElementaryFunctions
    & KeyPathIterable,
  Model.TangentVector.VectorSpaceScalar == Float

Pengoptimal RAdam.

Rectified Adam, varian Adam yang memperkenalkan istilah untuk memperbaiki varian kecepatan pembelajaran adaptif.

Referensi: “Tentang Varians Kecepatan Pembelajaran Adaptif dan Lebih Lanjutnya”

  • Pernyataan

    public typealias Model = Model
  • Kecepatan pembelajaran.

    Pernyataan

    public var learningRate: Float
  • Koefisien yang digunakan untuk menghitung momen gradien pertama dan kedua.

    Pernyataan

    public var beta1: Float
  • Koefisien yang digunakan untuk menghitung momen gradien pertama dan kedua.

    Pernyataan

    public var beta2: Float
  • Skalar kecil ditambahkan ke penyebut untuk meningkatkan stabilitas numerik.

    Pernyataan

    public var epsilon: Float
  • Penurunan kecepatan pembelajaran.

    Pernyataan

    public var decay: Float
  • Langkah saat ini.

    Pernyataan

    public var step: Int
  • Momen pertama Libra.

    Pernyataan

    public var firstMoments: Model.TangentVector
  • Momen kedua dari bobot.

    Pernyataan

    public var secondMoments: Model.TangentVector
  • Pernyataan

    public init(
      for model: __shared Model,
      learningRate: Float = 1e-3,
      beta1: Float = 0.9,
      beta2: Float = 0.999,
      epsilon: Float = 1e-8,
      decay: Float = 0
    )
  • Pernyataan

    public func update(_ model: inout Model, along direction: Model.TangentVector)
  • Pernyataan

    public required init(copying other: RAdam, to device: Device)