vahidk/tfrecord

Issue with shuffle=True, when using with pytorch dataloader

anjith2006 opened this issue · 2 comments

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

TFRecordDataset and MultiTFRecordDataset both shuffle the data internally. Provide a "shuffle_queue_size" to the dataset.

I added an example in the Readme.md