RMSProp

public struct RMSProp<Layer> : Optimizer where Layer : LayerType

extension RMSProp: Codable where Layer: Codable

Root mean square optimizer


                    
                    
                    ParamTensor

Undocumented

Swift

public typealias ParamTensor = Tensor<Layer.Parameter, Layer.Device>


                    
                    
                    model

Swift

public private(set) var model: Layer { get }


                    
                    
                    init(model:learningRate:gamma:epsilon:)

Root mean square optimizer

Swift

public init(model: Layer, learningRate: ParamTensor = 0.001, gamma: ParamTensor = 0.9, epsilon: ParamTensor = 1e-8)

`model`	Model to optimize
`learningRate`	Learning rate scaling factor
`gamma`	Exponential decay rate for gradient history
`epsilon`	Normalization scalar added to divisors


                    
                    
                    update(along:)

Swift

public mutating func update(along gradients: [ParamTensor])


                    
                    
                    init(from:)

Swift

public init(from decoder: Decoder) throws


                    
                    
                    encode(to:)

Swift

public func encode(to encoder: Encoder) throws