Bu sayfa, Cloud Translation API ile çevrilmiştir.

Konu Durumu

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

Bir cihazdaki eğitim döngüsünün durumu.

sınıflandırıcı
beyan
public var classifier: Model
optimize edici
beyan
public var optimizer: Opt

init(model:optimizer:id:devices:useAutomaticMixedPrecision :)

beyan

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

run(train:test:crossReplicaSumDevices:scheduleLearningRate:lossFunction :)

beyan

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