fix stupid mistakes on GPU multiple eval sets fallback
@@ -38,9 +38,11 @@ public:
Y_UNUSED(allowClearPool);
CB_ENSURE(testPoolPtrs.size() <= 1, "Multiple eval sets not supported for GPU");
Y_VERIFY(evalResultPtrs.size() == testPoolPtrs.size());
- NCatboostCuda::TrainModel(params, outputOptions, learnPool, *testPoolPtrs[0], model);
+ NCatboostCuda::TrainModel(params, outputOptions, learnPool, testPoolPtrs.size() ? *testPoolPtrs[0] : TPool(), model);
+ if (evalResultPtrs.size()) {
evalResultPtrs[0]->GetRawValuesRef().resize(model->ObliviousTrees.ApproxDimension);
}
+ }
void TrainModel(const NCatboostOptions::TPoolLoadParams& poolLoadParams,
const NCatboostOptions::TOutputFilesOptions& outputOptions,
