A detailed example of data loaders with PyTorch
stanford.edu › ~shervine › blogNow, when the sample corresponding to a given index is called, the generator executes the __getitem__ method to generate it. def __getitem__ ( self, index ): 'Generates one sample of data' # Select sample ID = self .list_IDs [index] # Load data and get label X = torch.load ( 'data/' + ID + '.pt' ) y = self .labels [ ID ] return X, y
python - How does the __getitem__'s idx work within ...
https://stackoverflow.com/questions/5883433813.11.2019 · I'm currently trying to use PyTorch's DataLoader to process data to feed into my deep learning model, but am facing some difficulty. The data that I need is of shape (minibatch_size=32, rows=100, columns=41).The __getitem__ code that I have within the custom Dataset class that I wrote looks something like this:. def __getitem__(self, idx): x = np.array(self.train.iloc[idx:100, :]) …