From ffebd812d1a69f42d102a821cd7ba36d187231e1 Mon Sep 17 00:00:00 2001 From: longyu Date: Sat, 8 Oct 2022 09:23:06 +0200 Subject: [PATCH] take less gpu memory to 20% --- user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py b/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py index 7882aca4a..cc39cec47 100644 --- a/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py +++ b/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py @@ -30,7 +30,7 @@ class CatboostPredictionBinaryMultiModel(BaseClassifierModel): cbr = CatBoostClassifier( allow_writing_files=False, - gpu_ram_part=0.5, + gpu_ram_part=0.2, verbose=100, early_stopping_rounds=400, **self.model_training_parameters,