Python hydra库(OmegaConf)(yaml)

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