torch.util.data.IterableDataset

[wangzeyu135798]

Hi:
In graph_dataset.py , class LoadBalanceGraphDataset(torch.utils.data.IterableDataset), self.num_samples default is 2000. I want to learn what's the relation between this variable and tensor.batch_size.In my own experiment, my class Dataset belong to IterableDataset, when self.num_samples is not the multiple of train_loader.batch_size, it will be wring.

[qibinc]

Hi @wangzeyu135798 ,

Please provide more information (e.g., python traceback) of the error. From your description, adding the drop_last=True param to the dataloader construction might solve the problem.