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 )