diff --git a/src_cpp/opennnBridge/nerlWorkerOpenNN.cpp b/src_cpp/opennnBridge/nerlWorkerOpenNN.cpp index 9502fbb9a..a269fa1f8 100644 --- a/src_cpp/opennnBridge/nerlWorkerOpenNN.cpp +++ b/src_cpp/opennnBridge/nerlWorkerOpenNN.cpp @@ -124,7 +124,7 @@ namespace nerlnet } - void NerlWorkerOpenNN::post_predict_process(fTensor2DPtr result_ptr){ + void NerlWorkerOpenNN::post_predict_process(fTensor2DPtr &result_ptr){ switch(_model_type){ case MODEL_TYPE_NN: { diff --git a/src_cpp/opennnBridge/nerlWorkerOpenNN.h b/src_cpp/opennnBridge/nerlWorkerOpenNN.h index 9e64e0fd8..16d40e339 100644 --- a/src_cpp/opennnBridge/nerlWorkerOpenNN.h +++ b/src_cpp/opennnBridge/nerlWorkerOpenNN.h @@ -31,7 +31,7 @@ class NerlWorkerOpenNN : public NerlWorker std::shared_ptr get_training_strategy_ptr() { return _training_strategy_ptr; }; std::shared_ptr get_data_set() { return _data_set; }; void post_training_process(fTensor2DPtr TrainDataNNptr); - void post_predict_process(fTensor2DPtr result_ptr); + void post_predict_process(fTensor2DPtr &result_ptr); void get_result_calc(fTensor2DPtr calculate_res,int num_of_samples,int inputs_number,fTensor2DPtr predictData); void set_optimization_method(int optimizer_type ,int learning_rate); void set_loss_method(int loss_method);