Shuffling videos. If only_keep_videos_from_0th_frame is False (default for gym envs), the videos in the dataset are HIGHLY correlated (they only differ in one frame) which makes the training highly unstable. This shuffling is not the best solution, but rather a hacky solution to this issue. Ideally, we want a full shuffling on the whole dataset.
Shuffling videos. If only_keep_videos_from_0th_frame is False (default for gym envs), the videos in the dataset are HIGHLY correlated (they only differ in one frame) which makes the training highly unstable. This shuffling is not the best solution, but rather a hacky solution to this issue. Ideally, we want a full shuffling on the whole dataset.

PiperOrigin-RevId: 206369208

tensorflow/tensor2tensor,vthorsteinsson/tensor2tensor,tensorflow/tensor2tensor,vthorsteinsson/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,vthorsteinsson/tensor2tensor,vthorsteinsson/tensor2tensor