From 94714fe6addff41e4984ff510f70cc415a131725 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 15 Apr 2023 00:51:17 +0200 Subject: [PATCH] Update requirements.txt to comment tensorboard (#11362) * Update requirements.txt comment tensorboard Signed-off-by: Glenn Jocher * Make tensorboard optional * Update __init__.py Signed-off-by: Glenn Jocher --------- Signed-off-by: Glenn Jocher --- requirements.txt | 2 +- utils/loggers/__init__.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index baf7a27..e1d98a8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -18,7 +18,7 @@ tqdm>=4.64.0 # protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 # Logging --------------------------------------------------------------------- -tensorboard>=2.4.1 +# tensorboard>=2.4.1 # clearml>=1.2.0 # comet diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index d428cdb..9b4c1d1 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -9,7 +9,6 @@ from pathlib import Path import pkg_resources as pkg import torch -from torch.utils.tensorboard import SummaryWriter from utils.general import LOGGER, colorstr, cv2 from utils.loggers.clearml.clearml_utils import ClearmlLogger @@ -20,6 +19,11 @@ from utils.torch_utils import de_parallel LOGGERS = ('csv', 'tb', 'wandb', 'clearml', 'comet') # *.csv, TensorBoard, Weights & Biases, ClearML RANK = int(os.getenv('RANK', -1)) +try: + from torch.utils.tensorboard import SummaryWriter +except ImportError: + SummaryWriter = lambda *args: None # None = SummaryWriter(str) + try: import wandb