Changing Imagenet dataset in examples to use a seed for file shuffling to achieve determinism.
Changing Imagenet dataset in examples to use a seed for file shuffling to achieve determinism.

PiperOrigin-RevId: 449095098

deepmind/kfac-jax,deepmind/kfac-jax