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