Python hydra库(OmegaConf)(yaml)

hxxjxw

已于 2023-03-09 13:52:51 修改

阅读量4.8k

点赞数 4

文章标签：深度学习

于 2023-03-06 15:56:54 首次发布

本文链接：https://blog.csdn.net/hxxjxw/article/details/129364417

版权

hydra简单来讲，就是更加方便地管理yaml config配置文件的

hydra使用OmegaConf这个库，用来解析yaml配置文件
pip install hydra-core
示例

config.yaml是这样
device: 'cpu'
dataset:
  name: imagenet
  path: ./dataset/imagenet
TTA: False
import hydra
from omegaconf import DictConfig

@hydra.main(config_name='config.yaml')
def my_app(cfg: DictConfig):
    #此时cfg就是一个dict
    print(cfg)
    print(cfg.dataset)
    print(cfg.dataset.name)
    
if __name__ == '__main__':
    my_app()
然后hydra运行时，会自动建立一个输出文件夹，包含日期和时间信息，然后还会直接将路径调到里面去，以方便保存脚本内的各种东西。

使用hydra的一个好处是我们可以在命令行非常方便的更改配置参数

例如上面代码运行命令是
python try.py
而我们想更改数据集，就可以直接
python try.py dataset=MNIST