PyTorch 데이터 파이프라인 구축: Dataset과 DataLoader의 구조 및 최적화
PyTorch에서 효율적인 데이터 파이프라인을 구축하기 위해서는 Dataset과 DataLoader의 역할을 명확히 이해해야 합니다.
Dataset은 데이터 저장소이자 인덱싱 매커니즘입니다. 원본 데이터의 위치를 정의하고, 특정 인덱스에 해당하는 단일 샘플을 추출 및 전처리하는 로직을 담당합니다.
DataLoader는 배치 조립 및 전송 파이프라인입니다. Dataset으로부터 샘플을 가져 ...
6월 12일 17:27에 게시됨