take less gpu memory to 20%

2022-10-08 09:23:06 +02:00
parent f0550f94d0
commit ffebd812d1
1 changed files with 1 additions and 1 deletions
--- 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,