File size: 858 Bytes
9c6750a
 
6045345
 
 
 
e2e68c3
 
 
6045345
 
7019509
 
6045345
 
 
 
 
 
f4df266
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
"""Module for wandb utilities"""

import os


def setup_wandb_env_vars(cfg):
    if cfg.wandb_mode and cfg.wandb_mode == "offline":
        os.environ["WANDB_MODE"] = cfg.wandb_mode
    elif cfg.wandb_project and len(cfg.wandb_project) > 0:
        os.environ["WANDB_PROJECT"] = cfg.wandb_project
        cfg.use_wandb = True
        if cfg.wandb_entity and len(cfg.wandb_entity) > 0:
            os.environ["WANDB_ENTITY"] = cfg.wandb_entity
        if cfg.wandb_watch and len(cfg.wandb_watch) > 0:
            os.environ["WANDB_WATCH"] = cfg.wandb_watch
        if cfg.wandb_log_model and len(cfg.wandb_log_model) > 0:
            os.environ["WANDB_LOG_MODEL"] = cfg.wandb_log_model
        if cfg.wandb_run_id and len(cfg.wandb_run_id) > 0:
            os.environ["WANDB_RUN_ID"] = cfg.wandb_run_id
    else:
        os.environ["WANDB_DISABLED"] = "true"