Commit 1c16df5a authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

set init_score_ = nullptr after delete

parent b3e99b27
...@@ -124,6 +124,7 @@ void Metadata::CheckOrPartition(data_size_t num_all_data, const std::vector<data ...@@ -124,6 +124,7 @@ void Metadata::CheckOrPartition(data_size_t num_all_data, const std::vector<data
if (init_score_ != nullptr && num_init_score_ != num_data_) { if (init_score_ != nullptr && num_init_score_ != num_data_) {
delete[] init_score_; delete[] init_score_;
Log::Error("Initial score size doesn't equal to data, score file will be ignored"); Log::Error("Initial score size doesn't equal to data, score file will be ignored");
init_score_ = nullptr;
num_init_score_ = 0; num_init_score_ = 0;
} }
} else { } else {
...@@ -148,6 +149,7 @@ void Metadata::CheckOrPartition(data_size_t num_all_data, const std::vector<data ...@@ -148,6 +149,7 @@ void Metadata::CheckOrPartition(data_size_t num_all_data, const std::vector<data
Log::Error("Initial score size doesn't equal to data , initial scores will be ignored"); Log::Error("Initial score size doesn't equal to data , initial scores will be ignored");
delete[] init_score_; delete[] init_score_;
num_init_score_ = 0; num_init_score_ = 0;
init_score_ = nullptr;
} }
// get local weights // get local weights
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment