Cannot find ResNet SAM checkpoints

google-research / vision_transformer

Apache License 2.0

10.02k stars 1.26k forks source link

Cannot find ResNet SAM checkpoints #217

Open yechao-zhang opened 2 years ago

yechao-zhang commented 2 years ago

As described in Readme file, the ResNet, ViT, and MLP-Mixer checkpoints optimized using "Surrogate Gap Minimization Improves Sharpness-Aware Training" are provided. However, I can only find ViT-SAM, ViT-GSAM, MLP-Mixer-SAM, MLP-Mixer-GSAM checkpoints in gs://vit_models and gs://mixer_models, respectively.

Where can I find the ResNet checkpoints to reproduce the results in SAM and GSAM papers?

yechao-zhang commented 2 years ago

I have found resnet checkpoints in gs://vit_models/gsam. It would be much appropriate providing ResNet-SAM checkpoints with different rho as well.