18 lines
685 B
Python
18 lines
685 B
Python
"""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_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
|