Issue with shuffle=True, when using with pytorch dataloader
anjith2006 opened this issue · 2 comments
anjith2006 commented
Hi,
Thank you for the very useful repo.
ValueError: DataLoader with IterableDataset: expected unspecified shuffle option, but got shuffle=True
is there a way to use shuffling with the interface. Also is there a way to preceompute
the length of the dataset.
Thanks
vahidk commented
TFRecordDataset and MultiTFRecordDataset both shuffle the data internally. Provide a "shuffle_queue_size" to the dataset.
vahidk commented
I added an example in the Readme.md