# https://github.com/thu-uav/OmniDrones/blob/dev/omni_drones/utils/wandb.py
# MIT License
#
# Copyright (c) 2023 Botian Xu, Tsinghua University
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
import wandb
from omegaconf import DictConfig, OmegaConf
import datetime
[docs]
def dict_flatten(a: dict, delim: str = "."):
"""Flatten a dict recursively.
"""
result = {}
for k, v in a.items():
if isinstance(v, dict):
result.update({k + delim + kk: vv for kk, vv in dict_flatten(v).items()})
else:
result[k] = v
return result
[docs]
def init_wandb(cfg: DictConfig):
wandb_cfg: DictConfig = cfg.wandb
kwargs = dict(
project=wandb_cfg.project,
group=wandb_cfg.group,
name=wandb_cfg.get("name", None),
mode=wandb_cfg.get("mode", "disabled"),
)
kwargs["id"] = wandb.util.generate_id()
run = wandb.init(**kwargs)
cfg_dict = dict_flatten(OmegaConf.to_container(cfg))
run.config.update(cfg_dict)
return run