[train] move set data set pointer into perform training in nerlworker…

…OpenNN
leondavi · Jul 7, 2024 · 4f38079 · 4f38079
1 parent f556b49
commit 4f38079
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
@@ -27,6 +27,8 @@ namespace nerlnet
 
     void NerlWorkerOpenNN::perform_training()
     {
+        this->_training_strategy_ptr->set_data_set_pointer(this->_data_set.get());
+
         TrainingResults res = this->_training_strategy_ptr->perform_training();
         this->_last_loss = res.get_training_error();
 

diff --git a/src_cpp/opennnBridge/openNNnif.cpp b/src_cpp/opennnBridge/openNNnif.cpp
@@ -24,8 +24,6 @@ void* trainFun(void* arg)
     nerlworker_opennn->set_dataset(data_set_ptr, TrainNNptr->data);
     data_set_ptr = nerlworker_opennn->get_data_set();
     // perform training
-    std::shared_ptr<TrainingStrategy> training_strategy_ptr = nerlworker_opennn->get_training_strategy_ptr();
-    training_strategy_ptr->set_data_set_pointer(nerlworker_opennn->get_dataset_ptr().get());
     nerlworker_opennn->perform_training();
     // post training
     nerlworker_opennn->post_training_process(TrainNNptr->data);