Switch to deep ensembles of size 3 for consistency with BatchEnsemble based methods.
Switch to deep ensembles of size 3 for consistency with BatchEnsemble based methods.

PiperOrigin-RevId: 447524891

google/uncertainty-baselines