데이터 로딩 & 전처리 파이프라인
model/data_loader.py는 데이터 로딩과 전처리를 담당하는 파이썬 스크립트다. 이 파일은 CIFAR-10 데이터셋을 사용하며, 데이터를 로딩하고 전처리하는 두 가지 주요 함수를 포함하고 있다.
fetch_dataloader(types, params)
•
인자:
◦
types: 'train' 또는 'dev' 문자열을 받아, 훈련 데이터셋 또는 검증 데이터셋을 로딩한다.
◦
params: 하이퍼파라미터 정보를 담고 있는 객체. 배치 크기, 데이터 증강 여부, CPU/GPU 사용 여부 등을 포함한다.
•
기능: CIFAR-10 데이터셋을 로딩하고, 데이터 증강을 선택적으로 적용한다.
•
반환값: DataLoader 객체를 반환한다.
•
train.py 에서 호출된다.