diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py index ab2c20d..f520fbb 100644 --- a/utils/loggers/wandb/wandb_utils.py +++ b/utils/loggers/wandb/wandb_utils.py @@ -19,16 +19,17 @@ from utils.datasets import LoadImagesAndLabels from utils.datasets import img2label_paths from utils.general import check_dataset, check_file +RANK = int(os.getenv('RANK', -1)) + try: import wandb assert hasattr(wandb, '__version__') # verify package import not local dir - if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2'): + if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2') and RANK in [0, -1]: wandb.login(timeout=30) except (ImportError, AssertionError): wandb = None -RANK = int(os.getenv('RANK', -1)) WANDB_ARTIFACT_PREFIX = 'wandb-artifact://'