Unverified 提交 1172ac15 authored 作者: Ayush Chaurasia's avatar Ayush Chaurasia 提交者: GitHub

W&B: DDP fix (#5176)

* fix dpp with wandb disabled * PyCharm reformat * Fix * fix Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 e520e820
...@@ -19,6 +19,13 @@ from utils.datasets import LoadImagesAndLabels ...@@ -19,6 +19,13 @@ from utils.datasets import LoadImagesAndLabels
from utils.datasets import img2label_paths from utils.datasets import img2label_paths
from utils.general import check_dataset, check_file from utils.general import check_dataset, check_file
try:
import wandb
assert hasattr(wandb, '__version__') # verify package import not local dir
except (ImportError, AssertionError):
wandb = None
RANK = int(os.getenv('RANK', -1)) RANK = int(os.getenv('RANK', -1))
WANDB_ARTIFACT_PREFIX = 'wandb-artifact://' WANDB_ARTIFACT_PREFIX = 'wandb-artifact://'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论