1 .創(chuàng)作靈感

小伙伴們再閱讀深度學(xué)習(xí)模型的代碼的時候，經(jīng)常會遇到y(tǒng)aml格式的配置文件。用這個配置文件是因為我們在訓(xùn)練模型的時候會涉及很多的參數(shù)，如果這些參數(shù)東一個，西一個，我們調(diào)起來的時候就會很不方便，所以用yaml腳本將他們集中到一起，類似的配置文件還有，json等。

2.yaml文本的基本格式

新建confg.yaml文本編寫以下內(nèi)容：

# 配置文件
manual_seed: 1device: cpu
model:name: Unetinchanlel: 1out_chanlel: 1f_maps: 64trainer:checkpoint_dir: '/home/user'resume: '/home/user/last_checkpoint.pytorch'

注意兩點：

（1）yaml通過鍵值對存儲信息，通過冒號和空格區(qū)分鍵和值

（2）不同級別用空格進(jìn)行縮進(jìn)，同一級別的內(nèi)容空格縮進(jìn)相同

3.讀取yaml文件的內(nèi)容

import yaml
# 讀取ymal文件# 打開文件
with open('config.yaml', 'r', encoding="utf-8") as f:#通過yaml.safe_load(f)函數(shù)讀取內(nèi)容config = yaml.safe_load(f)#通過字典索引的方式調(diào)用內(nèi)容
checkpoint_dir = config['trainer']['checkpoint_dir']print(f"地址: {checkpoint_dir}")

4. 寫入yaml文件

# 寫入yaml文件,以字典的形式存入數(shù)據(jù)
config_new = {'optimizer': {'learning_rate': 0.002,'weight_decay': 0.0001}}with open('config_new.yaml', 'w+') as f:yaml.safe_dump(config_new, f)

數(shù)據(jù)會被存入到新的yaml文件中

參考文章

玩轉(zhuǎn)PyYAML：讓Python與YAML格式數(shù)據(jù)無縫對接 - 知乎 (zhihu.com)