From 84be09ace943fee2108f3e2f61ee297888a39d38 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 21 Aug 2022 17:05:12 +0200 Subject: [PATCH 1/2] De-thread TensorBoard graph logging Issues with Classification models Signed-off-by: Glenn Jocher --- utils/loggers/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 98a123eee74d..904ce08ac055 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -292,7 +292,6 @@ def log_model(self, model_path, epoch=0, metadata={}): wandb.log_artifact(art) -@threaded def log_tensorboard_graph(tb, model, imgsz=(640, 640)): # Log model graph to TensorBoard try: From b31b9730627c84626f6ba2f91c749f0424a198b5 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 21 Aug 2022 17:05:34 +0200 Subject: [PATCH 2/2] Update __init__.py Signed-off-by: Glenn Jocher --- utils/loggers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 904ce08ac055..006125edbcd9 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -11,7 +11,7 @@ import torch from torch.utils.tensorboard import SummaryWriter -from utils.general import colorstr, cv2, threaded +from utils.general import colorstr, cv2 from utils.loggers.clearml.clearml_utils import ClearmlLogger from utils.loggers.wandb.wandb_utils import WandbLogger from utils.plots import plot_images, plot_labels, plot_results