I see there is a BufferedShuffleDataset in the source code: Add ShuffleDataset with buffer (#45290) · pytorch/pytorch@96540e9 · GitHub

However, I cannot find this back in the documentation and it seems not to be in release 1.7.1. When does this become available in the standard torch pip package?

1 Like

This PR was not cherry-picked into 1.7.1 and should be already available in the nightly binaries (docs) and the next stable minor release (1.8.0).

In can’t see it in recent PyTorch versions, has it been dropped ?