थ्रेडस्टेट

public var classifier: Model

public var optimizer: Opt

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

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