可以使用yaml文件存储训练参数,只需要使用pyyaml包load进来,就会将yaml文件转换为Python dict。
创建yaml文件
base_ckp_path: './check_point/'
datasets_path: './datasets/'
model:
save_acc: 0.
batch_size: 16
train_epochs: 200
validate_period: 10
lr: !!float 1e-3
lr_scheduler: cosine
在Python中调用
import yaml
f=open('config.yml')
cont=f.read()
yaml.load(cont,Loader=yaml.FullLoader)
需要注意的是,在yaml.load函数中,需要指定Loader为FullLoader,否则会报错。