SwanLab载入omegaconf配置教程

林泽毅2024-05-31 11:16

OmegaConf 是一个用于处理配置的 Python 库，尤其适用于需要灵活配置和配置合并的场景。

OmegaConf 与swanlab的集成非常简单，直接将omegaconf对象传递给swanlab.config，即可记录为超参数：

python 复制代码

from omegaconf import OmegaConf
import swanlab

cfg = OmegaConf.load("config.yaml")
swanlab.init(config=cfg,)

如果传递cfg时出现意外的结果，那么可以先转换omegaconf.DictConfig为原始类型：

python 复制代码

from omegaconf import OmegaConf
import swanlab

cfg = OmegaConf.load("config.yaml")
swanlab.init(config=OmegaConf.to_container(cfg, resolve=True, throw_on_missing=True))