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: [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?)