The size of my pickle file is 19700x4
I’m trying to work with 32-batch, but it does not divide the size of the pickle (row), so it takes all the rows, e.g: (32, 19700, 4). and it needs to be (32,4).
I don’t know where I’m wrong I think I need to define the sampler.
code:
class ROboDataset(Dataset):
def init(self, root, path, train=True, transform=None):
self.root = root
self.path = path
self.train = train
self.transform = transform
def __getitem__(self, index):
    root = self.root
    path = self.path
    data = pickle.load(file=open(os.path.join(root, path), "rb"))


    features = data[0]
    target = data[1]

    if self.transform is not None:
        features = self.transform(features)

    return features, target

def __len__(self):
    data = pickle.load(file=open(os.path.join(self.root, self.path), "rb"))
    return len(data[0])