I have a dataset “ds”. If I call
I get a return of
But if I do
train_loader = DataLoader(dataset=ds, batch_size=64, shuffle=True, num_workers=0)
samples = next(iter(train_loader))
I get one sample from the dataset. Not even an array of shape [sample] either - just the first sample.
What am I doing wrong? I’ve worked with datasets plenty in the past and they seemed to work fine.