ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 07:35:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Task.init() auto_connect_frameworks={'tfdefines':False} to allow disabling TF defines (issue #408)

Browse Source
This commit is contained in:
allegroai 2021-08-13 10:58:01 +03:00
parent 257a4058ee
commit 0168ba62e3
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/task.py
View File

@ -357,7 +357,7 @@ class Task(_Task):
auto_connect_frameworks={ auto_connect_frameworks={
'matplotlib': True, 'tensorflow': True, 'tensorboard': True, 'pytorch': True, 'matplotlib': True, 'tensorflow': True, 'tensorboard': True, 'pytorch': True,
'xgboost': True, 'scikit': True, 'fastai': True, 'lightgbm': True, 'xgboost': True, 'scikit': True, 'fastai': True, 'lightgbm': True,
'hydra': True, 'detect_repository': True, 'hydra': True, 'detect_repository': True, 'tfdefines': True,
} }
:param bool auto_resource_monitoring: Automatically create machine resource monitoring plots :param bool auto_resource_monitoring: Automatically create machine resource monitoring plots
@ -551,7 +551,9 @@ class Task(_Task):
PatchedMatplotlib.update_current_task(Task.__main_task) PatchedMatplotlib.update_current_task(Task.__main_task)
if is_auto_connect_frameworks_bool or auto_connect_frameworks.get('tensorflow', True) \ if is_auto_connect_frameworks_bool or auto_connect_frameworks.get('tensorflow', True) \
or auto_connect_frameworks.get('tensorboard', True): or auto_connect_frameworks.get('tensorboard', True):
PatchAbsl.update_current_task(Task.__main_task) # allow to disable tfdefines
if is_auto_connect_frameworks_bool or auto_connect_frameworks.get('tfdefines', True):
PatchAbsl.update_current_task(Task.__main_task)
TensorflowBinding.update_current_task( TensorflowBinding.update_current_task(
task, task,
patch_reporting=(is_auto_connect_frameworks_bool patch_reporting=(is_auto_connect_frameworks_bool