可以使用yaml文件存储训练参数,只需要使用pyyaml包load进来,就会将yaml文件转换为Python dict。

创建yaml文件

  1. base_ckp_path: './check_point/'
  2. datasets_path: './datasets/'
  3. model:
  4. save_acc: 0.
  5. batch_size: 16
  6. train_epochs: 200
  7. validate_period: 10
  8. lr: !!float 1e-3
  9. lr_scheduler: cosine

在Python中调用

  1. import yaml
  2. f=open('config.yml')
  3. cont=f.read()
  4. yaml.load(cont,Loader=yaml.FullLoader)

需要注意的是,在yaml.load函数中,需要指定Loader为FullLoader,否则会报错。