Update imagenet/hetsngp.py defaults to match the experiments used in the paper.
Update imagenet/hetsngp.py defaults to match the experiments used in the paper.

PiperOrigin-RevId: 410330707

google/uncertainty-baselines