PyYAML 을 설치하여 yaml(.yml or .yaml)파일을 다룰 수 있다
$ pip install PyYAML
## ./config.yam ##
GPU: '0'
VERBOSE: True
# Optimization arguments.
OPTIM:
BATCH_SIZE: 16
NUM_EPOCHS: 400 # 3000 # iteration
TRAINING:
VAL_AFTER_EVERY: 20
이런 yaml파일이 있을 때
with open('./config.yaml', 'r') as f:
cfg = yaml.load(f, ***Loader=yaml.FullLoader***)
이렇게 하면 dictionary 형태로 불러오게된다
{’GPU’: ’0’, ‘VERBOSE’:True, ‘OPTIM’ : {’BATCH_SIZE’ : 16, {NUM_EPOCHES:40}}
저 Loader부분이 궁금했다
Loader=yaml.FullLoader :Loader=yaml.SafeLoader :