가끔 데이터를 쓰다보면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를 가지는 데이터만을 데이터셋으로 불러올 수 있다.