ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-23 01:55:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix failed catboost bind on GPU (#592)

Browse Source
This commit is contained in:
allegroai 2022-03-06 19:33:11 +02:00
parent ac1750b442
commit e142954bf4
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
clearml/binding/frameworks/catboost_bind.py
View File

@ -39,10 +39,9 @@ class PatchCatBoostModelIO(PatchBaseModelIO):
CatBoostClassifier.fit = _patched_call(CatBoostClassifier.fit, PatchCatBoostModelIO._fit) CatBoostClassifier.fit = _patched_call(CatBoostClassifier.fit, PatchCatBoostModelIO._fit)
CatBoostRegressor.fit = _patched_call(CatBoostRegressor.fit, PatchCatBoostModelIO._fit) CatBoostRegressor.fit = _patched_call(CatBoostRegressor.fit, PatchCatBoostModelIO._fit)
CatBoostRanker.fit = _patched_call(CatBoostRegressor.fit, PatchCatBoostModelIO._fit) CatBoostRanker.fit = _patched_call(CatBoostRegressor.fit, PatchCatBoostModelIO._fit)
except ImportError: except Exception as e:
pass logger = PatchCatBoostModelIO.__main_task.get_logger()
except Exception: logger.report_text("Failed patching Catboost. Exception is: '" + str(e) + "'")
pass
@staticmethod @staticmethod
def _save(original_fn, obj, f, *args, **kwargs): def _save(original_fn, obj, f, *args, **kwargs):
@ -94,6 +93,16 @@ class PatchCatBoostModelIO(PatchBaseModelIO):
def _fit(original_fn, obj, *args, **kwargs): def _fit(original_fn, obj, *args, **kwargs):
callbacks = kwargs.get("callbacks") or [] callbacks = kwargs.get("callbacks") or []
kwargs["callbacks"] = callbacks + [PatchCatBoostModelIO.__callback_cls(task=PatchCatBoostModelIO.__main_task)] kwargs["callbacks"] = callbacks + [PatchCatBoostModelIO.__callback_cls(task=PatchCatBoostModelIO.__main_task)]
# noinspection PyBroadException
try:
return original_fn(obj, *args, **kwargs)
except Exception:
logger = PatchCatBoostModelIO.__main_task.get_logger()
logger.report_text(
"Catboost metrics logging is not supported for GPU. "
"See https://github.com/catboost/catboost/issues/1792"
)
del kwargs["callbacks"]
return original_fn(obj, *args, **kwargs) return original_fn(obj, *args, **kwargs)
@staticmethod @staticmethod