Trạng thái Chủ đề

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

Trạng thái của vòng lặp huấn luyện trên thiết bị.

• sự phân loại
  Tuyên ngôn

  public var classifier: Model

• trình tối ưu hóa
  Tuyên ngôn

  public var optimizer: Opt

• init(model:optimizer:id:devices:useAutomaticMixedPrecision:)
  Tuyên ngôn

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

• run(train:test:crossReplicaSumDevices:scheduleLearningRate:lossFunction:)
  Tuyên ngôn

  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>)