ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 09:07:00 +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 ..import_bind import PostImportHookPatching
from ...debugging.log import LoggerRoot
from .tensorflow_bind import IsTensorboardInit
try:
import fastai
@ -51,7 +52,7 @@ class PatchFastaiV1(object):
@staticmethod
def patch_model_callback():
# 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
try:
@ -191,7 +192,7 @@ class PatchFastaiV2(object):
@staticmethod
def patch_model_callback():
if "tensorboard" in sys.modules:
if "tensorboard" in sys.modules and IsTensorboardInit.tensorboard_used():
return
# noinspection PyBroadException