mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

De-thread TensorBoard graph logging (#9071)

Browse Source 
* De-thread TensorBoard graph logging

Issues with Classification models

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2022-08-21 17:07:56 +02:00 committed by GitHub
parent 5373a28c1b
commit db2ee5fa0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/loggers/__init__.py
View File

@ -11,7 +11,7 @@ import pkg_resources as pkg
import torch import torch
from torch.utils.tensorboard import SummaryWriter 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.clearml.clearml_utils import ClearmlLogger
from utils.loggers.wandb.wandb_utils import WandbLogger from utils.loggers.wandb.wandb_utils import WandbLogger
from utils.plots import plot_images, plot_labels, plot_results from utils.plots import plot_images, plot_labels, plot_results
@ -292,7 +292,6 @@ class GenericLogger:
wandb.log_artifact(art) wandb.log_artifact(art)
@threaded
def log_tensorboard_graph(tb, model, imgsz=(640, 640)): def log_tensorboard_graph(tb, model, imgsz=(640, 640)):
# Log model graph to TensorBoard # Log model graph to TensorBoard
try: try: