MLCompute_MNIST_TrainingGraph.swift

private func buildTrainingGraph() {
    trainingGraph = MLCTrainingGraph(graphObjects: [graph],
                                         lossLayer: MLCLossLayer(descriptor: MLCLossDescriptor(type: .softmaxCrossEntropy,
                                                                                               reductionType: .mean)),
                                         optimizer: MLCAdamOptimizer(descriptor: MLCOptimizerDescriptor(learningRate: 0.001,
                                                                                                       gradientRescale: 1.0,
                                                                                                    regularizationType: .none,
                                                                                                    regularizationScale: 0.0),
                                                                     beta1: 0.9,
                                                                     beta2: 0.999,
                                                                     epsilon: 1e-7,
                                                                     timeStep: 1))

    trainingGraph.addInputs(["image" : inputTensor],
                            lossLabels: ["label" : lossLabelTensor])

    trainingGraph.compile(options: [], device: device)
}