RefineThenCalibrate-Vision

Code to run the computer vision benchmark in the paper "Rethinking Early Stopping: Refine, Then Calibrate" https://arxiv.org/abs/2501.19195.

Files

main.py: Launch runs and log results.
utils.py: Contains our pytorch-lightning module that allows benchmarking TS-Refinement against other early stopping metrics.
figures.ipynb: Generate figures for the paper.
resnet.py and wide_resnet.py: Deep learning model used, from https://github.com/uoguelph-mlrg/Cutout

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
README.md		README.md
figures.ipynb		figures.ipynb
main.py		main.py
resnet.py		resnet.py
utils.py		utils.py
wide_resnet.py		wide_resnet.py