You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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.
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.
The text was updated successfully, but these errors were encountered: