가끔 데이터를 쓰다보면torch.util.data.Dataset을 list를 통해 indexing 하고 싶을 때가 있을것이다.
이를 위해torch.utils.data.Subset이 존재한다.
train_dataset = Subset(train_dataset, list1)이런식으로 사용하면, 내가 원하는 index를 가지는 데이터만을 데이터셋으로 불러올 수 있다.
거인의 어깨에 올라서서 더 넓은 세상을 바라보라 - 아이작 뉴턴
가끔 데이터를 쓰다보면torch.util.data.Dataset을 list를 통해 indexing 하고 싶을 때가 있을것이다.
이를 위해torch.utils.data.Subset이 존재한다.
train_dataset = Subset(train_dataset, list1)이런식으로 사용하면, 내가 원하는 index를 가지는 데이터만을 데이터셋으로 불러올 수 있다.