728x90 반응형 Dataset1 [PyTorch] Dataset and DataLoader Dataset 이란 PyTorch 의 tensor 와 학습에 사용될 일반 raw data (흔히, storage에 저장된 파일들) 사이에 위치하며, raw-data로부터 PyTorch의 기본데이터형 인 tensor를 얻게 해주는 역할을 수행하는 class임. 일반적으로 task에 따라 다양한 형태의 raw-data가 있기 때문에, 각 task별로 Custom Dataset을 만드는 경우가 잦음. Custom Dataset 만들기 torch.util.data 모듈의 Dataset을 상속하고, 다음의 methods를 overriding해야만 함. __init__(self) : Dataset 인스턴스에 대한 생성자로 데이터셋에 대한 초기화를 담당. raw-data에 따라 parameters를 자유롭게 추가할 .. 2024. 4. 9. 이전 1 다음 728x90 반응형