cd to repo's root and use dl_model_ckpt.sh to download most of the checkpoints.
The checkpoints of other models need to be downloaded manually. Rename files and put them under exp/weights
| model | link | filename under exp/weights |
|---|---|---|
| CutMix | link | cutmix_r50.pth |
| Cutout | link | cutout_r50.pth |
| AugMix | link | augmix_r50.pth |
| Uniform Soup | link | uniform_soup_vit-b.pth |
| RobustViT | link | robust_vit.pth |
Finally, for Greedy Soup, follow model-soups to generate greedy soup checkpoint and move it to exp/weights/greedy_soup_vit-b.pth.