SGDOptimizer
public struct SGDOptimizer : Optimizer, OptimizerAddons
Stochastic gradient descent optimizer.
-
Declaration
Swift
public let graph: DynamicGraph -
Declaration
Swift
public var nesterov: Bool -
Declaration
Swift
public var rate: Float -
Declaration
Swift
public var scale: Float -
Declaration
Swift
public var decay: Float -
Declaration
Swift
public var momentum: Float -
Declaration
Swift
public var dampening: Float -
Declaration
Swift
public var parameters: [any DynamicGraph_AnyParameters] { get set } -
Declaration
Swift
public init( _ graph: DynamicGraph, nesterov: Bool, rate: Float, scale: Float, decay: Float, momentum: Float, dampening: Float ) -
Declaration
Swift
public func step(streamContext: StreamContext?)
SGDOptimizer Structure Reference