[2주차(수)] Pytorch - Dataset, Dataloader

과제 푸는 데 시간이 너무 많이 걸려서 며칠동안 글 쓰는 걸 빼먹었다.. 밀린 걸 다 쓰려니 또 너무 많아서 그냥 오늘 배운 내용 위주로 먼저 써야겠다.. Dataset의 기본 구성 요소 torch.utils.data에서 Dataset 클래스를 상속해서 만든다. 크게 __init__, __len__, __getitem__ 메서드로 구성된다. from torch.utils.data import Dataset class CustomDataset(Dataset): def __init__(self, a, b): # 초기 데이터 생성 방법, 데이터의 위치, 파일명 등 초기화 작업 self.a = a self.b = b def __len__(self): # Dataset의 전체 길이 반환 return len(self.a) def __getitem__(self, idx): # Dataset의 idx번째 데이터를 반환 return self.a[idx], self.b[idx] DataLoader의 ...

원문링크 : [2주차(수)] Pytorch - Dataset, Dataloader