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
Python
복사
이런 yaml파일이 있을 때
with open('./config.yaml', 'r') as f:
cfg = yaml.load(f, Loader=yaml.FullLoader)
Python
복사
이렇게 하면 dictionary 형태로 불러오게된다
{’GPU’: ’0’,
‘VERBOSE’:True,
‘OPTIM’ :
{’BATCH_SIZE’ : 16, {NUM_EPOCHES:40}}
저 Loader부분이 궁금했다
•
Loader=yaml.FullLoader :
•
Loader=yaml.SafeLoader :
