حالة الخيط

public class ThreadState<Model: Layer, Opt: Optimizer>
where
  Opt.Model == Model, Opt.Scalar == Float, Model.Input == Tensor<Float>,
  Model.Output == Tensor<Float>,
  Model.TangentVector.VectorSpaceScalar == Float

حالة حلقة التدريب على الجهاز.

• المصنف
  تصريح

  public var classifier: Model

• محسن
  تصريح

  public var optimizer: Opt

• init(model:optimizer:id:devices:useAutomaticMixedPrecision :)
  تصريح

  public init(
    model: Model, optimizer: Opt, id: Int, devices: [Device], useAutomaticMixedPrecision: Bool
  )

• تشغيل (القطار:اختبار:crossReplicaSumDevices:scheduleLearningRate:lossFunction :)
  تصريح

  public func run<Dataset: Sequence>(
    train: Dataset, test: Dataset, crossReplicaSumDevices: [Device]? = nil,
    scheduleLearningRate: (Opt) -> Void = { _ in },
    lossFunction: @differentiable (Tensor<Float>, @noDerivative Tensor<Int32>) -> Tensor<Float> =
      _defaultLossFunction
  )
    -> () -> (train: HostStatistics, test: HostStatistics)
  where Dataset.Iterator.Element == (x: Tensor<Float>, y: Tensor<Int32>)