Predict fix

leondavi · Jul 8, 2024 · 17fb979 · 17fb979
1 parent 9bb7c09
commit 17fb979
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -31,7 +31,7 @@ class NerlWorkerOpenNN : public NerlWorker
     std::shared_ptr<opennn::TrainingStrategy> get_training_strategy_ptr() { return _training_strategy_ptr; };
     std::shared_ptr<opennn::DataSet> 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);