Trang này được dịch bởi Cloud Translation API.

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