From fb96998f54868c65ba0a8ae1ecd7f8f25582e0bf Mon Sep 17 00:00:00 2001 From: Hoyt Koepke Date: Thu, 1 Oct 2020 16:57:10 -0700 Subject: [PATCH] Updates to MLCompute interface to use correct API. --- src/ml/neural_net/mlc_dc_backend.mm | 2 +- src/ml/neural_net/mlc_layer_weights.mm | 2 +- src/ml/neural_net/mlc_od_backend.mm | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/ml/neural_net/mlc_dc_backend.mm b/src/ml/neural_net/mlc_dc_backend.mm index 5a4a0d0cc6..712d1cb0b2 100644 --- a/src/ml/neural_net/mlc_dc_backend.mm +++ b/src/ml/neural_net/mlc_dc_backend.mm @@ -154,7 +154,7 @@ mlc_layer_weights import_weights(const float_array_map &weights) // Instantiate training graph. MLCGraph *graph = [MLCGraph tc_graphForDrawingClassifierDescriptor:descriptor batchSize:batch_size]; - training_graph_ = [MLCTrainingGraph trainingGraphWithGraphObjects:@[ graph ] + training_graph_ = [MLCTrainingGraph graphWithGraphObjects:@[ graph ] lossLayer:loss optimizer:optimizer]; diff --git a/src/ml/neural_net/mlc_layer_weights.mm b/src/ml/neural_net/mlc_layer_weights.mm index 0a03ed6fe7..adc928514c 100644 --- a/src/ml/neural_net/mlc_layer_weights.mm +++ b/src/ml/neural_net/mlc_layer_weights.mm @@ -195,7 +195,7 @@ MLCTensorData *optimizerData1 = copy_float_array(optimizer_data_1_key, optimizer_data_1); MLCTensorData *optimizerData2 = copy_float_array(optimizer_data_2_key, optimizer_data_2); if (tensors_[layerName] != nil) { - BOOL success = [tensors_[layerName] optimizerData:@[ optimizerData1, optimizerData2 ]]; + BOOL success = [tensors_[layerName] bindOptimizerData:@[ optimizerData1, optimizerData2] deviceData:nil]; if (!success) { throw("Failed to load optimizer data into tensor"); } diff --git a/src/ml/neural_net/mlc_od_backend.mm b/src/ml/neural_net/mlc_od_backend.mm index cd5990fdfa..cc678d40a2 100644 --- a/src/ml/neural_net/mlc_od_backend.mm +++ b/src/ml/neural_net/mlc_od_backend.mm @@ -166,10 +166,10 @@ mlc_layer_weights import_weights(const float_array_map &weights) MLCSGDOptimizer *optimizer = [MLCSGDOptimizer optimizerWithDescriptor:create_optimizer_descriptor(config) momentumScale:0.9f - usesNestrovMomentum:NO]; + usesNesterovMomentum:NO]; // Instantiate training graph. - result.trainingGraph = [MLCTrainingGraph trainingGraphWithGraphObjects:@[ graph ] + result.trainingGraph = [MLCTrainingGraph graphWithGraphObjects:@[ graph ] lossLayer:lossLayer optimizer:optimizer];