Enable to run OOD evaluation on the SNGP model on CIFAR during training without the need for a saved checkpoint.
Enable to run OOD evaluation on the SNGP model on CIFAR during training without the need for a saved checkpoint.

PiperOrigin-RevId: 396540519

google/uncertainty-baselines