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 fastAI binding not reporting scalars if tensorboard is installed

Browse Source
This commit is contained in:
Alex Burlacu 2023-08-24 13:55:05 +03:00
parent 0b521b00a6
commit 0959974920
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml/binding/frameworks/fastai_bind.py
View File

@ -8,6 +8,7 @@ from . import _patched_call
from .tensorflow_bind import WeightsGradientHistHelper from .tensorflow_bind import WeightsGradientHistHelper
from ..import_bind import PostImportHookPatching from ..import_bind import PostImportHookPatching
from ...debugging.log import LoggerRoot from ...debugging.log import LoggerRoot
from .tensorflow_bind import IsTensorboardInit
try: try:
import fastai import fastai
@ -51,7 +52,7 @@ class PatchFastaiV1(object):
@staticmethod @staticmethod
def patch_model_callback(): def patch_model_callback():
# if you have tensorboard, we assume you use TensorboardLogger, which we catch, so no need to patch. # if you have tensorboard, we assume you use TensorboardLogger, which we catch, so no need to patch.
if "tensorboard" in sys.modules: if "tensorboard" in sys.modules and IsTensorboardInit.tensorboard_used():
return return
try: try:
@ -191,7 +192,7 @@ class PatchFastaiV2(object):
@staticmethod @staticmethod
def patch_model_callback(): def patch_model_callback():
if "tensorboard" in sys.modules: if "tensorboard" in sys.modules and IsTensorboardInit.tensorboard_used():
return return
# noinspection PyBroadException # noinspection PyBroadException