SmoothL1Loss
public struct SmoothL1Loss : Loss
Smooth L1 loss (for object detection).
-
Declaration
Swift
public var beta: Float
-
Declaration
Swift
public init(beta: Float = 1)
-
Declaration
Swift
public func callAsFunction<T: DynamicGraph.AnyTensorGroup, U: DynamicGraph.AnyTensorGroup>( _ input: T, target: U, streamContext: StreamContext? ) -> [T.AnyTensor] where T.AnyTensor == U.AnyTensor