public struct ParameterGroupOptimizerBuilder
एक ParameterGroupOptimizer
बनाता है। इसका उपयोग अनिवार्य रूप से मॉडल में एकल भार के स्तर पर किया जाता है। ( [Bool]
से पैरामीटरग्रुपऑप्टिमाइज़र) द्वारा चयनित पैरामीटर समूहों से मैपिंग अंतिम ऑप्टिमाइज़र को परिभाषित करती है।
घोषणा
public init()
घोषणा
public mutating mutating func makeParameter(_ name: String, _ value: Float) -> GlobalAccessor
घोषणा
public subscript(global: GlobalAccessor) -> Float { get }
घोषणा
public subscript(state name: String) -> StateAccessor { mutating get }
घोषणा
public subscript(local name: String) -> LocalAccessor { mutating get }
कॉलबैक को कॉलबैक की सूची में जोड़ता है।
घोषणा
public mutating mutating func appendCallback(_ cb: @escaping OptimizerCallback)
ऑप्टिमाइज़र लौटाता है और बिल्डर को साफ़ करता है।
घोषणा
public mutating mutating func makeOptimizer() -> ParameterGroupOptimizer
वर्तमान पैरामीटर समूह अनुकूलन के लिए गति के साथ एक sgdStep लागू करता है।
घोषणा
public mutating func sgdStep( nesterov: Bool, mom: GlobalAccessor, lr: GlobalAccessor, velocity: StateAccessor )
क्लिप्ड ट्रस्ट अनुपात (एलएआरएस में प्रयुक्त) की गणना करता है।
घोषणा
public mutating func clippedTrustRatio( trustCoefficient: GlobalAccessor, epsilon: GlobalAccessor, weightDecay: GlobalAccessor ) -> LocalAccessor
ट्रस्टरेशियो (LARS में प्रयुक्त) द्वारा ग्रेडिएंट को मापता है।
घोषणा
public mutating mutating func scaleGradByTrustRatio(trustRatio: LocalAccessor)
ग्रेडिएंट पर भार क्षय स्केलिंग लागू करता है।
घोषणा
public mutating mutating func scaleGradient(byWeightDecay weightDecay: GlobalAccessor)
नई ग्रेडिएंट (सीखने की दर के आधार पर स्केल) के आधार पर वेग पैरामीटर की पुन: गणना करता है।
घोषणा
public mutating func updateVelocity( mom: GlobalAccessor, lr: GlobalAccessor, velocity: StateAccessor )