GradScaler

Swift

public var scale: Float

Swift

public var growthFactor: Float

Swift

public var backoffFactor: Float

Swift

public var growthInterval: Int

Swift

public init(
  scale: Float = 65536, growthFactor: Float = 2, backoffFactor: Float = 0.5,
  growthInterval: Int = 2_000
)

Swift

public func scale<T: DynamicGraph.TensorGroup>(_ loss: T, streamContext: StreamContext? = nil)
  -> T

Swift

public mutating func unscale<T>(_ optimizers: [T], streamContext: StreamContext? = nil) where T : Optimizer

Swift

public mutating func step<T: Optimizer>(
  _ optimizers: inout [T], streamContext: StreamContext? = nil
)

Swift

public mutating func unscale<T>(_ optimizer: T, streamContext: StreamContext? = nil) where T : Optimizer

Swift

public mutating func step<T>(_ optimizer: inout T, streamContext: StreamContext? = nil) where T : Optimizer