Adds a README note on tf eager execution mode for tfds dataset loading.
Adds a README note on tf eager execution mode for tfds dataset loading.

PiperOrigin-RevId: 402644600

google/uncertainty-baselines