Unverified 提交 db2ee5fa authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

De-thread TensorBoard graph logging (#9071)

* De-thread TensorBoard graph logging Issues with Classification models Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * Update __init__.py Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 5373a28c
......@@ -11,7 +11,7 @@ import pkg_resources as pkg
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
......@@ -292,7 +292,6 @@ class GenericLogger:
wandb.log_artifact(art)
@threaded
def log_tensorboard_graph(tb, model, imgsz=(640, 640)):
# Log model graph to TensorBoard
try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论