Robust segvit compatibility + new dataset.#1183

ekellbuch wants to merge 170 commits into

google:maingoogle/uncertainty-baselines:mainfrom

ekellbuch:run_openekellbuch/uncertainty-baselines:run_openCopy head branch name to clipboard