adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

take less gpu memory to 20%

Browse Source
This commit is contained in:
longyu 2022-10-08 09:23:06 +02:00
parent f0550f94d0
commit ffebd812d1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py
View File

@ -30,7 +30,7 @@ class CatboostPredictionBinaryMultiModel(BaseClassifierModel):
cbr = CatBoostClassifier( cbr = CatBoostClassifier(
allow_writing_files=False, allow_writing_files=False,
gpu_ram_part=0.5, gpu_ram_part=0.2,
verbose=100, verbose=100,
early_stopping_rounds=400, early_stopping_rounds=400,
**self.model_training_parameters, **self.model_training_parameters,