public struct ParameterGroupOptimizerBuilder
یک ParameterGroupOptimizer
می سازد. این اساساً در سطح یک وزن در مدل استفاده می شود. نقشه برداری از گروه های پارامتر انتخاب شده توسط ( [Bool]
به ParameterGroupOptimizer) بهینه ساز نهایی را مشخص می کند.
اعلامیه
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 )
ClippedTrustRatio را محاسبه می کند (در LARS استفاده می شود).
اعلامیه
public mutating func clippedTrustRatio( trustCoefficient: GlobalAccessor, epsilon: GlobalAccessor, weightDecay: GlobalAccessor ) -> LocalAccessor
گرادیان را توسط trustRatio (در 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 )