Hydra使用

最简单使用

安装

1
pip install hydra-core

初步测试

目前的代码结构

1
2
3
4
folder
├── conf
│   └── recoro_train.yaml
└── main.py

其中main.py 

1
2
3
4
5
6
7
8
9
10
11
12
13
from omegaconf import DictConfig, OmegaConf
import hydra

@hydra.main(config_path="./conf", config_name="recoro_train")
def main(cfg: DictConfig):
    running_dir = str(hydra.utils.get_original_cwd())
    working_dir = str(Path.cwd())
    print(f"The current running directory is {running_dir}")
    print(f"The current working directory is {working_dir}")


if __name__ == "__main__":
    main()

结果

1
2
The current running directory is C:\Users\xx\xx\xx\folder\
The current working directory is C:\Users\xx\xx\xx\folder\outputs\2023-02-08\22-47-06

可以看到hydra运行时，会自动建立一个输出文件夹，包含日期和时间信息，然后还会直接将路径调到里面去，以方便保存脚本内的各种东西。这就是初步测试，全部都在这一行：配置的路径在”conf”，配置的文件名为”recoro_train”

因此，无论在代码的任何地方，新建文件都会在working directory下 进行新建写入。

【Python】Hydra 库使用记录_Kin__Zhang的博客-CSDN博客_python hydra